Update app.py

app.py

@@ -48,6 +48,7 @@ gmaps = googlemaps.Client(key=os.getenv("GOOGLE_API_KEY"))
 
 # Helper Functions
 def bag_of_words(s, words):
+    """Convert user input to bag-of-words vector."""
     bag = [0] * len(words)
     s_words = word_tokenize(s)
     s_words = [stemmer.stem(word.lower()) for word in s_words if word.isalnum()]
@@ -58,6 +59,7 @@ def bag_of_words(s, words):
     return np.array(bag)
 
 def generate_chatbot_response(message, history):
+    """Generate chatbot response and maintain conversation history."""
     history = history or []
     try:
         result = chatbot_model.predict([bag_of_words(message, words)])
@@ -73,6 +75,7 @@ def generate_chatbot_response(message, history):
     return history, response
 
 def analyze_sentiment(user_input):
+    """Analyze sentiment and map to emojis."""
     inputs = tokenizer_sentiment(user_input, return_tensors="pt")
     with torch.no_grad():
         outputs = model_sentiment(**inputs)
@@ -81,6 +84,7 @@ def analyze_sentiment(user_input):
     return f"Sentiment: {sentiment_map[sentiment_class]}"
 
 def detect_emotion(user_input):
+    """Detect emotions based on input."""
     pipe = pipeline("text-classification", model=model_emotion, tokenizer=tokenizer_emotion)
     result = pipe(user_input)
     emotion = result[0]["label"].lower().strip()
@@ -95,35 +99,42 @@ def detect_emotion(user_input):
     return emotion_map.get(emotion, "Unknown 🤔"), emotion
 
 def generate_suggestions(emotion):
+    """Return relevant suggestions based on detected emotions."""
     emotion_key = emotion.lower()
     suggestions = {
         "joy": [
-            ["Relaxation Techniques", '<a href="https://www.helpguide.org/mental-health/meditation" target="_blank">Visit</a>'],
-            ["Emotional Toolkit", '<a href="https://www.nih.gov" target="_blank">Visit</a>'],
-            ["Stress Management", '<a href="https://www.health.harvard.edu" target="_blank">Visit</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": [
-            ["Handle Anger", '<a href="https://youtu.be/MIc299Flibs" target="_blank">Watch</a>'],
-            ["Stress Tips", '<a href="https://www.helpguide.org/mental-health/anger-management.htm" target="_blank">Visit</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": [
-            ["Coping with Anxiety", '<a href="https://www.helpguide.org/mental-health/anxiety" target="_blank">Visit</a>'],
-            ["Mindfulness", '<a href="https://youtu.be/yGKKz185M5o" target="_blank">Watch</a>'],
+            ["Mindfulness Practices", '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Visit</a>'],
+            ["Coping with Anxiety", '<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/yGKKz185M5o" target="_blank">Watch</a>'],
         ],
         "sadness": [
-            ["Overcoming Sadness", '<a href="https://youtu.be/-e-4Kx5px_I" target="_blank">Watch</a>'],
+            ["Emotional Wellness Toolkit", '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Visit</a>'],
+            ["Dealing with Anxiety", '<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/-e-4Kx5px_I" target="_blank">Watch</a>'],
         ],
         "surprise": [
-            ["Managing Surprises", '<a href="https://www.health.harvard.edu" target="_blank">Visit</a>'],
+            ["Managing Stress", '<a href="https://www.health.harvard.edu/health-a-to-z" target="_blank">Visit</a>'],
+            ["Coping Strategies", '<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/m1vaUGtyo-A" target="_blank">Watch</a>'],
         ],
-        "neutral": [
-            ["General Well-Being Tips", '<a href="https://www.psychologytoday.com" target="_blank">Visit</a>'],
-        ],
     }
     return suggestions.get(emotion_key, [["No specific suggestions available.", ""]])
 
 def get_health_professionals_and_map(location, query):
+    """Search nearby healthcare professionals using Google Maps API."""
     try:
         if not location or not query:
             return ["Please provide both location and query."], ""
@@ -148,18 +159,18 @@ def get_health_professionals_and_map(location, query):
 
 # Main Application Logic
 def app_function(user_input, location, query, history):
-    chatbot_history, _ = generate_chatbot_response(user_input, history)
-    sentiment_result = analyze_sentiment(user_input)
-    emotion_result, cleaned_emotion = detect_emotion(user_input)
-    suggestions = generate_suggestions(cleaned_emotion)
-    professionals, map_html = get_health_professionals_and_map(location, query)
+    chatbot_history, _ = generate_chatbot_response(user_input, history)  # Generate chatbot response
+    sentiment_result = analyze_sentiment(user_input)  # Sentiment detection
+    emotion_result, cleaned_emotion = detect_emotion(user_input)  # Emotion detection
+    suggestions = generate_suggestions(cleaned_emotion)  # Generate suggestions based on emotion
+    professionals, map_html = get_health_professionals_and_map(location, query)  # Find nearby professionals and map
     return chatbot_history, sentiment_result, emotion_result, suggestions, professionals, map_html
 
 # Gradio Interface
 custom_css = """
 body {
     font-family: 'Roboto', sans-serif;
-    background: linear-gradient(135deg,#0d0d0d,#ff5722);
+    background: linear-gradient(135deg, #0d0d0d, #ff5722); /* Background gradient */
     color: white;
 }
 
@@ -174,48 +185,43 @@ h1 {
 }
 
 textarea, input {
-    background: black; /* Black background */
-    color: white; /* White text always active */
+    background: transparent;
+    color: black;
     border: 2px solid orange;
+    padding: 8px;
     font-size: 1rem;
-    padding: 10px;
+    caret-color: black;
+    outline: none;
     border-radius: 8px;
-    caret-color: white; /* White typing cursor */
-    outline: none; /* No focus outline */
 }
 
-/* Ensure the text stays white and background black when focused */
 textarea:focus, input:focus {
-    background: black;
-    color: white;
+    background: transparent;
+    color: black;
     border: 2px solid orange;
     outline: none;
 }
 
-/* Prevent hover effects from breaking consistency */
 textarea:hover, input:hover {
-    background: black;
-    color: white;
+    background: transparent;
+    color: black;
     border: 2px solid orange;
-    outline: none;
 }
 
-button {
-    background: linear-gradient(135deg, orange, #ff4500);
-    color: white;
+.df-container {
+    background: white;
+    color: black;
+    border: 2px solid orange;
+    border-radius: 10px;
     padding: 10px;
-    border-radius: 8px;
-    font-weight: bold;
-    font-size: 1.2rem;
-    border: none;
-    cursor: pointer;
-}
-
-button:hover {
-    box-shadow: 0px 4px 8px rgba(255, 165, 0, 0.5);
+    font-size: 14px;
+    max-height: 400px; /* Extends the height of the table */
+    height: auto;
+    overflow-y: auto; /* Adds scroll if content overflows */
 }
 """
 
+# Gradio Application
 with gr.Blocks(css=custom_css) as app:
     gr.HTML("<h1>🌟 Well-Being Companion</h1>")
     with gr.Row():
@@ -225,15 +231,15 @@ with gr.Blocks(css=custom_css) as app:
     chatbot = gr.Chatbot(label="Chat History")
     sentiment = gr.Textbox(label="Detected Sentiment")
     emotion = gr.Textbox(label="Detected Emotion")
-    suggestions = gr.DataFrame(headers=["Title", "Link"])
+    suggestions = gr.DataFrame(headers=["Title", "Link"])  # Table for suggestions
     professionals = gr.Textbox(label="Nearby Professionals", lines=6)
     map_html = gr.HTML(label="Interactive Map")
-    submit = gr.Button("Submit")
+    submit = gr.Button(value="Submit", variant="primary")
 
     submit.click(
         app_function,
         inputs=[user_input, location, query, chatbot],
-        outputs=[chatbot, sentiment, emotion, suggestions, professionals, map_html]
+        outputs=[chatbot, sentiment, emotion, suggestions, professionals, map_html],
     )
 
 app.launch()