Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import requests
 import torch
 import pandas as pd
+import folium
 from bs4 import BeautifulSoup
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 from nltk.tokenize import word_tokenize
@@ -83,7 +84,8 @@ def chat(message, history):
         response = f"An error occurred: {str(e)}"
     
     history.append((message, response))
-    return history, response
+    return history, history
+
 
 # Sentiment analysis setup
 tokenizer_sentiment = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
@@ -97,59 +99,51 @@ def load_emotion_model():
 
 tokenizer_emotion, model_emotion = load_emotion_model()
 
-# Emotion detection function with suggestions in plain English and resources in table
+# Emotion detection function with suggestions
 def detect_emotion(user_input):
     pipe = pipeline("text-classification", model=model_emotion, tokenizer=tokenizer_emotion)
     result = pipe(user_input)
     emotion = result[0]['label']
     
-    # Define emotion-specific message and resources
+    suggestions = []
+    video_link = ""
+    
+    # Provide suggestions based on the detected emotion
     if emotion == 'joy':
-        emotion_msg = "You're feeling happy! Keep up the great mood!"
-        resources = [
-            {"subject": "Relaxation Techniques", "heading": "Mindful Breathing Meditation", "link": "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation", "video_url": "https://youtu.be/m1vaUGtyo-A"},
-            {"subject": "Dealing with Stress", "heading": "Tips for Dealing with Anxiety", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety", "video_url": "https://youtu.be/m1vaUGtyo-A"},
-            {"subject": "Emotional Wellness Toolkit", "heading": "Emotional Wellness Resources", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit", "video_url": "https://youtu.be/m1vaUGtyo-A"}
+        suggestions = [
+            ("Relaxation Techniques", "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"),
+            ("Dealing with Stress", "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"),
+            ("Emotional Wellness Toolkit", "https://www.nih.gov/health-information/emotional-wellness-toolkit")
         ]
-        
+        video_link = "Watch on YouTube: https://youtu.be/m1vaUGtyo-A"
     elif emotion == 'anger':
-        emotion_msg = "You're feeling angry. It's okay to feel this way. Let's try to calm down."
-        resources = [
-            {"subject": "Emotional Wellness Toolkit", "heading": "Managing Emotions", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit", "video_url": "https://youtu.be/MIc299Flibs"},
-            {"subject": "Stress Management Tips", "heading": "Managing Stress Effectively", "link": "https://www.health.harvard.edu/health-a-to-z", "video_url": "https://youtu.be/MIc299Flibs"},
-            {"subject": "Dealing with Anger", "heading": "Strategies to Calm Anger", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety", "video_url": "https://youtu.be/MIc299Flibs"}
+        suggestions = [
+            ("Emotional Wellness Toolkit", "https://www.nih.gov/health-information/emotional-wellness-toolkit"),
+            ("Stress Management Tips", "https://www.health.harvard.edu/health-a-to-z"),
+            ("Dealing with Anger", "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety")
         ]
-        
+        video_link = "Watch on YouTube: https://youtu.be/MIc299Flibs"
     elif emotion == 'fear':
-        emotion_msg = "You're feeling fearful. Take a moment to breathe and relax."
-        resources = [
-            {"subject": "Mindfulness Practices", "heading": "Breathing Techniques", "link": "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation", "video_url": "https://youtu.be/yGKKz185M5o"},
-            {"subject": "Coping with Anxiety", "heading": "Overcoming Fear", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety", "video_url": "https://youtu.be/yGKKz185M5o"},
-            {"subject": "Emotional Wellness Toolkit", "heading": "Calming Your Mind", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit", "video_url": "https://youtu.be/yGKKz185M5o"}
+        suggestions = [
+            ("Mindfulness Practices", "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"),
+            ("Coping with Anxiety", "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"),
+            ("Emotional Wellness Toolkit", "https://www.nih.gov/health-information/emotional-wellness-toolkit")
         ]
-        
+        video_link = "Watch on YouTube: https://youtu.be/yGKKz185M5o"
     elif emotion == 'sadness':
-        emotion_msg = "You're feeling sad. It's okay to take a break."
-        resources = [
-            {"subject": "Emotional Wellness Toolkit", "heading": "Restoring Your Emotional Health", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit", "video_url": "https://youtu.be/-e-4Kx5px_I"},
-            {"subject": "Dealing with Anxiety", "heading": "Coping Strategies for Stress", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety", "video_url": "https://youtu.be/-e-4Kx5px_I"}
+        suggestions = [
+            ("Emotional Wellness Toolkit", "https://www.nih.gov/health-information/emotional-wellness-toolkit"),
+            ("Dealing with Anxiety", "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety")
         ]
-        
+        video_link = "Watch on YouTube: https://youtu.be/-e-4Kx5px_I"
     elif emotion == 'surprise':
-        emotion_msg = "You're feeling surprised. It's okay to feel neutral!"
-        resources = [
-            {"subject": "Managing Stress", "heading": "Relaxation Tips", "link": "https://www.health.harvard.edu/health-a-to-z", "video_url": "https://youtu.be/m1vaUGtyo-A"},
-            {"subject": "Coping Strategies", "heading": "Dealing with Unexpected Events", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety", "video_url": "https://youtu.be/m1vaUGtyo-A"}
+        suggestions = [
+            ("Managing Stress", "https://www.health.harvard.edu/health-a-to-z"),
+            ("Coping Strategies", "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety")
         ]
-        
-    else:
-        emotion_msg = "Could not detect emotion."
-        resources = []
-    
-    # Create a DataFrame for resources to display in table format
-    resource_df = pd.DataFrame(resources)
+        video_link = "Watch on YouTube: https://youtu.be/m1vaUGtyo-A"
     
-    return emotion_msg, resource_df
+    return emotion, suggestions, video_link
 
 # Google Geocoding API setup to convert city name to latitude/longitude
 geocode_url = "https://maps.googleapis.com/maps/api/geocode/json"
@@ -168,86 +162,90 @@ def get_lat_lon(location, api_key):
             return location['lat'], location['lng']
     return None, None
 
-# Google Places API setup for wellness professionals
-url = "https://maps.googleapis.com/maps/api/place/textsearch/json"
-places_details_url = "https://maps.googleapis.com/maps/api/place/details/json"
-api_key = os.getenv("GOOGLE_API_KEY")  # Use environment variable for security
-
-# Function to get places data using Google Places API
-def get_places_data(query, location, radius, api_key, next_page_token=None):
-    params = {
-        'query': query,
-        'location': location,
-        'radius': radius,
-        'key': api_key
-    }
-    if next_page_token:
-        params['pagetoken'] = next_page_token
-    
-    response = requests.get(url, params=params)
-    return response.json()
-
-# Function to fetch wellness professionals
+# Get wellness professionals
 def get_wellness_professionals(location, api_key):
+    query = "therapist OR counselor OR mental health professional OR marriage and family therapist OR psychotherapist OR psychiatrist OR psychologist OR nutritionist OR wellness doctor OR holistic practitioner OR integrative medicine OR chiropractor OR naturopath"
+    radius = 50000  # 50 km radius
+
+    # Get the latitude and longitude from the location input
     lat, lon = get_lat_lon(location, api_key)
     
-    if lat and lon:
-        places = get_places_data("wellness professional", f"{lat},{lon}", 10000, api_key)
-        
-        if places and 'results' in places:
-            professionals = []
-            for place in places['results']:
-                name = place.get("name", "No name available")
-                rating = place.get("rating", "No rating available")
-                address = place.get("formatted_address", "No address available")
-                professionals.append({
-                    "Name": name,
-                    "Rating": rating,
-                    "Address": address
-                })
-                
-            professionals_df = pd.DataFrame(professionals)
-            return professionals_df
-        else:
-            return "No wellness professionals found nearby."
-    else:
-        return "Location not found. Please check the location."
+    if lat is None or lon is None:
+        return "Unable to find coordinates for the given location."
+
+    # Using Google Places API to fetch wellness professionals
+    data = get_places_data(query, f"{lat},{lon}", radius, api_key)
+
+    if data:
+        results = data.get('results', [])
+        wellness_data = []
+        for place in results:
+            name = place.get("name")
+            address = place.get("formatted_address")
+            latitude = place.get("geometry", {}).get("location", {}).get("lat")
+            longitude = place.get("geometry", {}).get("location", {}).get("lng")
+            wellness_data.append([name, address, latitude, longitude])
+        return wellness_data
+
+    return []
+
+# Function to generate a map with wellness professionals
+def generate_map(wellness_data):
+    map_center = [23.685, 90.3563]  # Default center for Bangladesh (you can adjust this)
+    m = folium.Map(location=map_center, zoom_start=12)
+
+    for place in wellness_data:
+        name, address, lat, lon = place
+        folium.Marker(
+            location=[lat, lon],
+            popup=f"<b>{name}</b><br>{address}",
+            icon=folium.Icon(color='blue', icon='info-sign')
+        ).add_to(m)
+
+    # Save map as an HTML file
+    map_file = "wellness_map.html"
+    m.save(map_file)
+    
+    # Return the HTML file path to be embedded in Gradio
+    return map_file
 
-# Gradio interface function to handle actions and outputs
-def interface_function(message, action, location, history):
-    history = history or []
+# Gradio interface setup for user interaction
+def user_interface(message, location, history, api_key):
+    history, history = chat(message, history)
     
-    if action == "Chat":
-        # Use chat function if 'Chat' button is clicked
-        history, response = chat(message, history)
-    elif action == "Detect Emotion":
-        # Use emotion detection if 'Detect Emotion' button is clicked
-        emotion_msg, resource_df = detect_emotion(message)
-        response = emotion_msg
-        # Return the resource DataFrame as a table
-        return history, response, resource_df
-    elif action == "Wellness Resources":
-        # Use location to get wellness professionals if 'Wellness Resources' is clicked
-        if not location.strip():
-            response = "Please enter a valid location."
-        else:
-            professionals_df = get_wellness_professionals(location, api_key)
-            if isinstance(professionals_df, pd.DataFrame):
-                response = "Found wellness professionals nearby:"
-                return history, response, professionals_df
-            else:
-                response = professionals_df  # If error message is returned
-                return history, response, None
-
-    return history, "Invalid action", None
-
-# Gradio Interface with table outputs for emotion and wellness professionals
-iface = gr.Interface(
-    fn=interface_function,
-    inputs=["text", "radio", "text", "state"],  # Include state in the inputs
-    outputs=["text", "dataframe", "state"],  # Add state to the outputs
-    live=True,
-    allow_flagging="never"
+    # Sentiment analysis
+    inputs = tokenizer_sentiment(message, return_tensors="pt")
+    outputs = model_sentiment(**inputs)
+    sentiment = ["Negative", "Neutral", "Positive"][torch.argmax(outputs.logits, dim=1).item()]
+    
+    # Emotion detection
+    emotion, resources, video_link = detect_emotion(message)
+    
+    # Get wellness professionals
+    wellness_data = get_wellness_professionals(location, api_key)
+    
+    # Generate the map
+    map_file = generate_map(wellness_data)
+    
+    # Create a DataFrame for the suggestions
+    suggestions_df = pd.DataFrame(resources, columns=["Subject", "Article URL"])
+    suggestions_df["Video URL"] = video_link  # Add video URL column
+    
+    return history, history, sentiment, emotion, resources, video_link, map_file, suggestions_df.to_html(escape=False)
+
+# Gradio chatbot interface
+chatbot = gr.Chatbot(label="Mental Health Chatbot")
+location_input = gr.Textbox(label="Enter your location (latitude,longitude)", placeholder="e.g., 21.3,-157.8")
+
+# Gradio interface definition
+demo = gr.Interface(
+    user_interface,
+    [gr.Textbox(label="Message"), location_input, "state", "text"],
+    [chatbot, "state", "text", "text", "json", "text", "html", "html"],  # Added additional output for the map
+    allow_flagging="never",
+    title="Mental Health & Well-being Assistant"
 )
 
-iface.launch(share=True)
+# Launch Gradio interface
+if __name__ == "__main__":
+    demo.launch()