Update app.py

app.py

@@ -14,6 +14,7 @@ import folium
 import os
 import base64
 import torch  # Added missing import for torch
+from PIL import Image
 
 # Disable GPU usage for TensorFlow
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
@@ -101,106 +102,75 @@ def detect_emotion(user_input):
     emotion = result[0]['label']
     return f"Emotion Detected: {emotion}"
 
-# Initialize Google Maps API client securely
-gmaps = googlemaps.Client(key=os.getenv('GOOGLE_API_KEY'))
-
-# Function to search for health professionals
-def search_health_professionals(query, location, radius=10000):
-    places_result = gmaps.places_nearby(location, radius=radius, type='doctor', keyword=query)
-    return places_result.get('results', [])
-
-# Function to get directions and display on Gradio UI
-def get_health_professionals_and_map(current_location, health_professional_query):
-    try:
-        location = gmaps.geocode(current_location)
-        if not location:
-            return "Location not found. Please try again with a valid location.", ""
-        lat = location[0]["geometry"]["location"]["lat"]
-        lng = location[0]["geometry"]["location"]["lng"]
-        location = (lat, lng)
-        
-        professionals = search_health_professionals(health_professional_query, location)
-        
-        # Generate map
-        map_center = location
-        m = folium.Map(location=map_center, zoom_start=13)
-        
-        # Add markers to the map
-        for place in professionals:
-            folium.Marker(
-                location=[place['geometry']['location']['lat'], place['geometry']['location']['lng']],
-                popup=place['name']
-            ).add_to(m)
-        
-        # Convert map to HTML for Gradio display
-        map_html = m._repr_html_()
-        
-        # Route information
-        route_info = "\n".join([f"{place['name']} - {place['vicinity']}" for place in professionals])
-        
-        return route_info, map_html
-    except googlemaps.exceptions.HTTPError as e:
-        return f"Error occurred with Google Maps API: {e}", ""
-    except Exception as e:
-        return f"Unexpected error: {str(e)}", ""
-
-# Function to generate suggestions based on the detected emotion
+# Function to generate suggestions based on detected emotion
 def generate_suggestions(emotion):
     suggestions = {
-        'joy': [
-            {"Title": "Relaxation Techniques", "Subject": "Relaxation", "Link": '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Mindful Breathing Meditation</a>'},
-            {"Title": "Dealing with Stress", "Subject": "Stress Management", "Link": '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Tips for Dealing with Anxiety</a>'},
-            {"Title": "Emotional Wellness Toolkit", "Subject": "Wellness", "Link": '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Emotional Wellness Toolkit</a>'},
-            {"Title": "Relaxation Video", "Subject": "Video", "Link": '<a href="https://youtu.be/m1vaUGtyo-A" target="_blank">Watch Video</a>'}
-        ],
-        'anger': [
-            {"Title": "Emotional Wellness Toolkit", "Subject": "Wellness", "Link": '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Emotional Wellness Toolkit</a>'},
-            {"Title": "Stress Management Tips", "Subject": "Stress Management", "Link": '<a href="https://www.health.harvard.edu/health-a-to-z" target="_blank">Harvard Health: Stress Management</a>'},
-            {"Title": "Dealing with Anger", "Subject": "Anger Management", "Link": '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Tips for Dealing with Anger</a>'},
-            {"Title": "Relaxation Video", "Subject": "Video", "Link": '<a href="https://youtu.be/MIc299Flibs" target="_blank">Watch Video</a>'}
-        ],
-        'fear': [
-            {"Title": "Mindfulness Practices", "Subject": "Mindfulness", "Link": '<a href="https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation" target="_blank">Mindful Breathing Meditation</a>'},
-            {"Title": "Coping with Anxiety", "Subject": "Anxiety Management", "Link": '<a href="https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety" target="_blank">Tips for Dealing with Anxiety</a>'},
-            {"Title": "Emotional Wellness Toolkit", "Subject": "Wellness", "Link": '<a href="https://www.nih.gov/health-information/emotional-wellness-toolkit" target="_blank">Emotional Wellness Toolkit</a>'},
-            {"Title": "Relaxation Video", "Subject": "Video", "Link": '<a href="https://youtu.be/yGKKz185M5o" target="_blank">Watch Video</a>'}
-        ]
+        'joy': ["Stay positive! Keep up the good mood.", "Try some relaxing activities like meditation."],
+        'anger': ["It's okay to be angry, try to breathe and relax.", "Exercise can help release tension."],
+        'fear': ["Take deep breaths, you are in control.", "Try mindfulness exercises to calm your mind."],
+        'sadness': ["Take a break, it's okay to feel down sometimes.", "Consider reaching out to a friend or loved one."],
+        'surprise': ["Take a moment to reflect, things might seem overwhelming.", "Practice mindfulness to regain balance."],
+        'disgust': ["It's okay to feel disgust, try to identify the cause.", "Taking a short walk might help clear your mind."]
     }
+    return pd.DataFrame(suggestions.get(emotion, ["Stay positive!"]))
+
+# Function to get nearby health professionals and create a map
+def get_health_professionals_and_map(location, health_professional_query):
+    # Use Google Maps API to get health professionals (example setup)
+    gmaps = googlemaps.Client(key="YOUR_GOOGLE_API_KEY")
+    places = gmaps.places(query=health_professional_query, location=location)
     
-    emotion_suggestions = suggestions.get(emotion, [])
-    return pd.DataFrame(emotion_suggestions)
+    if places['status'] == 'OK':
+        results = places['results']
+        route_info = "\n".join([place['name'] for place in results])
+        map_html = create_map(results)
+        return route_info, map_html
+    return "No professionals found.", None
+
+def create_map(places):
+    m = folium.Map(location=[places[0]['geometry']['location']['lat'], places[0]['geometry']['location']['lng']], zoom_start=13)
+    for place in places:
+        folium.Marker([place['geometry']['location']['lat'], place['geometry']['location']['lng']], 
+                      popup=place['name']).add_to(m)
+    map_html = m._repr_html_()
+    return map_html
 
 # Custom CSS styling for Gradio interface
 css = """
     body {
-        font-family: 'Arial', sans-serif;
+        font-family: 'Roboto', sans-serif;
     }
     .gradio-container {
-        background-color: #f7f7f7;
+        background-color: #f0f0f0;
+        font-size: 16px;
     }
     .gradio-input, .gradio-output {
-        padding: 10px;
-        border-radius: 5px;
-        background-color: #fff;
-        border: 1px solid #ccc;
+        padding: 15px;
+        border-radius: 10px;
+        background-color: #ffffff;
+        border: 2px solid #ccc;
     }
     .gradio-container .gradio-button {
-        background-color: #4CAF50;
+        background-color: #007BFF;
         color: white;
         border-radius: 5px;
+        padding: 10px 15px;
     }
     .gradio-container .gradio-button:hover {
-        background-color: #45a049;
-    }
-    .gradio-output {
-        font-size: 16px;
-        color: #333;
+        background-color: #0056b3;
     }
     .gradio-container h3 {
         color: #333;
     }
     .gradio-output .output {
-        border-top: 2px solid #ddd;
+        border-top: 3px solid #ddd;
+        padding-top: 10px;
+    }
+    .gradio-input input {
+        color: #333;
+    }
+    .gradio-input textarea {
+        color: #333;
     }
 """
 
@@ -216,7 +186,18 @@ def gradio_app(message, current_location, health_professional_query, history):
     # Get health professionals details and map
     route_info, map_html = get_health_professionals_and_map(current_location, health_professional_query)
     
-    return sentiment, emotion, suggestions_df, route_info, map_html, history
+    # Add emoticon for emotion
+    emotion_emoticons = {
+        'joy': '😊',
+        'anger': '😡',
+        'fear': '😨',
+        'sadness': '😢',
+        'surprise': '😲',
+        'disgust': '🤢'
+    }
+    emotion_icon = emotion_emoticons.get(emotion, '🙂')
+    
+    return sentiment, f"{emotion_icon} {emotion}", suggestions_df, route_info, map_html, history
 
 # Gradio interface setup
 iface = gr.Interface(
@@ -242,4 +223,4 @@ iface = gr.Interface(
 )
 
 # Launch Gradio interface
-iface.launch(share=True)
+iface.launch(share=True)