Update app.py

app.py

@@ -5,16 +5,12 @@ from datasets import load_dataset
 import markdown2
 import signal
 
-# Set Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
 
-# Initialize Flask app
 application = Flask(__name__, static_folder='static', template_folder='templates')
 
-# Get Hugging Face token from environment variables
 hf_token = os.getenv("HF_TOKEN")
 
-# Load datasets and inference client from Hugging Face Hub
 chat_doctor_dataset = load_dataset("avaliev/chat_doctor")
 mental_health_dataset = load_dataset("Amod/mental_health_counseling_conversations")
 
@@ -23,7 +19,6 @@ client = InferenceClient(
     token=hf_token,
 )
 
-# Function to select context based on the user's input
 def select_relevant_context(user_input):
     mental_health_keywords = [
         "anxious", "depressed", "stress", "mental health", "counseling", 
@@ -48,30 +43,31 @@ def select_relevant_context(user_input):
 
     return context
 
-# Function to create the prompt to send to the AI model
 def create_prompt(context, user_input):
-    return f"{context}\n\nUser: {user_input}\nAssistant:"
+    prompt = (
+        f"{context}\n\n"
+        f"User: {user_input}\nAssistant:"
+    )
+    return prompt
 
 # Function to render Markdown into HTML
 def render_markdown(text):
     return markdown2.markdown(text)
 
-# Flask route to serve the homepage
 @application.route('/')
 def index():
     return render_template('index.html')
 
-# Route to serve static files
 @application.route('/static/<path:path>')
 def send_static(path):
     return send_from_directory('static', path)
 
-# Route to handle chat interaction
 @application.route('/chat', methods=['POST'])
 def chat():
     user_input = request.json['message']
     
     context = select_relevant_context(user_input)
+    
     prompt = create_prompt(context, user_input)
     
     response = ""
@@ -86,7 +82,6 @@ def chat():
     
     return jsonify({"response": formatted_response})
 
-# Route to shutdown the server (if necessary)
 @application.route('/shutdown', methods=['POST'])
 def shutdown():
     if request.environ.get('werkzeug.server.shutdown'):
@@ -95,7 +90,6 @@ def shutdown():
         os.kill(os.getpid(), signal.SIGINT)
     return jsonify({"message": "Server is shutting down..."})
 
-# Function to handle the server shutdown
 def shutdown_server():
     func = request.environ.get('werkzeug.server.shutdown')
     if func is None:
@@ -103,6 +97,5 @@ def shutdown_server():
     else:
         func()
 
-# Run the app without specifying host and port (Hugging Face Spaces handles this)
 if __name__ == '__main__':
-    application.run()
+    application.run(host='0.0.0.0', port=7680)