Update application.py

application.py

@@ -3,6 +3,7 @@ from flask import Flask, request, jsonify, render_template, send_from_directory
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import markdown2
+import signal
 
 os.environ["HF_HOME"] = "/app/.cache"
 
@@ -28,13 +29,16 @@ def select_relevant_context(user_input):
         "pain", "illness", "disease", "infection", "surgery"
     ]
 
+    # Check if the input contains any mental health-related keywords
     if any(keyword in user_input.lower() for keyword in mental_health_keywords):
         example = mental_health_dataset['train'][0]
         context = f"Counselor: {example['Response']}\nUser: {example['Context']}"
+    # Check if the input contains any medical-related keywords
     elif any(keyword in user_input.lower() for keyword in medical_keywords):
         example = chat_doctor_dataset['train'][0]
         context = f"Doctor: {example['input']}\nPatient: {example['output']}"
     else:
+        # If no specific keywords are found, provide a general response
         context = "You are a general assistant. Respond to the user's query in a helpful manner."
 
     return context
@@ -46,6 +50,7 @@ def create_prompt(context, user_input):
     )
     return prompt
 
+# Function to render Markdown into HTML
 def render_markdown(text):
     return markdown2.markdown(text)
 
@@ -60,7 +65,9 @@ def send_static(path):
 @app.route('/chat', methods=['POST'])
 def chat():
     user_input = request.json['message']
+    
     context = select_relevant_context(user_input)
+    
     prompt = create_prompt(context, user_input)
     
     response = ""
@@ -75,5 +82,18 @@ def chat():
     
     return jsonify({"response": formatted_response})
 
-if __name__ == '__main__':
-    app.run(debug=False)
+@app.route('/shutdown', methods=['POST'])
+def shutdown():
+    if request.environ.get('werkzeug.server.shutdown'):
+        shutdown_server()
+    else:
+        os.kill(os.getpid(), signal.SIGINT)
+    return jsonify({"message": "Server is shutting down..."})
+
+def shutdown_server():
+    func = request.environ.get('werkzeug.server.shutdown')
+    if func is None:
+        os.kill(os.getpid(), signal.SIGINT)  # Kill the process if Werkzeug is not available
+    else:
+        func()
+