Update app.py

app.py

@@ -13,17 +13,17 @@ import googlemaps
 import folium
 import torch
 
-# Disable GPU usage for TensorFlow
+# Disable GPU usage for TensorFlow and suppress TensorFlow warnings
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # Suppress TensorFlow warnings
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
-# Ensure necessary NLTK resources are downloaded
+# Download NLTK resources
 nltk.download("punkt")
 
-# Initialize stemmer
+# Initialize Lancaster Stemmer
 stemmer = LancasterStemmer()
 
-# Load chatbot intents and training data
+# Load intents.json and training data for the chatbot
 with open("intents.json") as file:
     intents_data = json.load(file)
 
@@ -95,37 +95,21 @@ def detect_emotion(user_input):
 def generate_suggestions(emotion):
     suggestions = {
         "joy": [
-            ["Relaxation Techniques", '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Read</a>'],
-            ["Dealing with Stress", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Read</a>'],
-            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Read</a>'],
+            ["Relaxation Techniques", '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Visit</a>'],
+            ["Dealing with Stress", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Visit</a>'],
+            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Visit</a>'],
             ["Relaxation Video", '<a href="https://youtu.be/m1vaUGtyo-A" target="_blank">Watch</a>'],
         ],
         "anger": [
-            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Read</a>'],
-            ["Stress Management Tips", '<a href="https://www.health.harvard.edu/health-a-to-z" target="_blank">Read</a>'],
-            ["Dealing with Anger", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Read</a>'],
+            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Visit</a>'],
+            ["Stress Management Tips", '<a href="https://www.health.harvard.edu/health-a-to-z" target="_blank">Visit</a>'],
+            ["Dealing with Anger", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Visit</a>'],
             ["Relaxation Video", '<a href="https://youtu.be/MIc299Flibs" target="_blank">Watch</a>'],
         ],
-        "fear": [
-            ["Mindfulness Practices", '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Read</a>'],
-            ["Coping with Anxiety", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Read</a>'],
-            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Read</a>'],
-            ["Relaxation Video", '<a href="https://youtu.be/yGKKz185M5o" target="_blank">Watch</a>'],
-        ],
-        "sadness": [
-            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Read</a>'],
-            ["Dealing with Anxiety", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Read</a>'],
-            ["Relaxation Video", '<a href="https://youtu.be/-e-4Kx5px_I" target="_blank">Watch</a>'],
-        ],
-        "surprise": [
-            ["Managing Stress", '<a href="https://www.health.harvard.edu/health-a-to-z" target="_blank">Read</a>'],
-            ["Coping Strategies", '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Read</a>'],
-            ["Relaxation Video", '<a href="https://youtu.be/m1vaUGtyo-A" target="_blank">Watch</a>'],
-        ],
     }
-    return suggestions.get(emotion)
+    return suggestions.get(emotion, [["No suggestions available", "", ""]])
 
-# Search nearby professionals and generate map
+# Search professionals and generate map
 def get_health_professionals_and_map(location, query):
     try:
         geo_location = gmaps.geocode(location)
@@ -140,42 +124,43 @@ def get_health_professionals_and_map(location, query):
                 folium.Marker([place["geometry"]["location"]["lat"], place["geometry"]["location"]["lng"]],
                               popup=place["name"]).add_to(map_)
             return professionals, map_._repr_html_()
-        return [], ""
+        return ["No professionals found"], ""
     except Exception as e:
         return [f"Error: {e}"], ""
 
-# Application logic integrated in one function
+# Main app function
 def app_function(message, location, query, history):
     chatbot_history, _ = chatbot(message, history)
     sentiment = analyze_sentiment(message)
     emotion = detect_emotion(message.lower())
     suggestions = generate_suggestions(emotion)
-    professionals_info, map_html = get_health_professionals_and_map(location, query)
-    return chatbot_history, sentiment, emotion, suggestions, professionals_info, map_html
+    professionals, map_html = get_health_professionals_and_map(location, query)
+    return chatbot_history, sentiment, emotion, suggestions, professionals, map_html
 
 # Gradio app interface
 with gr.Blocks() as app:
     gr.Markdown("# 🌟 Well-Being Companion")
-    gr.Markdown("Empowering Your Mental Health Journey 💚")
+    gr.Markdown("Empowering your Well-Being journey 💚")
 
     with gr.Row():
         user_message = gr.Textbox(label="Your Message", placeholder="Enter your message...")
-        user_location = gr.Textbox(label="Your Location", placeholder="Enter location...")
-        search_query = gr.Textbox(label="Query (e.g., therapist)", placeholder="Search for professionals...")
-        submit_btn = gr.Button(value="Submit")
-
-    chatbot_box = gr.Chatbot(label="Chat History", type="messages")
-    emotion_display = gr.Textbox(label="Detected Emotion")
-    sentiment_display = gr.Textbox(label="Detected Sentiment")
-    suggestions_table = gr.DataFrame(headers=["Title", "Links"], label="Suggestions", height=250)
+        user_location = gr.Textbox(label="Your Location", placeholder="Enter your location...")
+        search_query = gr.Textbox(label="Query", placeholder="Search for professionals...")
+        submit_btn = gr.Button("Submit")
+
+    chatbot_box = gr.Chatbot(label="Chat History")
+    emotion_output = gr.Textbox(label="Detected Emotion")
+    sentiment_output = gr.Textbox(label="Detected Sentiment")
+    suggestions_output = gr.DataFrame(headers=["Title", "Links"], label="Suggestions")
     map_output = gr.HTML(label="Nearby Professionals Map")
-    professional_display = gr.Textbox(label="Nearby Professionals", lines=5)
+    professional_list = gr.Textbox(label="Nearby Professionals", lines=5)
 
     submit_btn.click(
         app_function,
         inputs=[user_message, user_location, search_query, chatbot_box],
         outputs=[
-            chatbot_box, sentiment_display, emotion_display, suggestions_table, professional_display, map_output,
+            chatbot_box, sentiment_output, emotion_output,
+            suggestions_output, professional_list, map_output,
         ],
     )