Update app.py

app.py

@@ -98,6 +98,64 @@ def load_emotion_model():
 
 tokenizer_emotion, model_emotion = load_emotion_model()
 
+# Emotion detection function with suggestions in plain English
+def detect_emotion(user_input):
+    pipe = pipeline("text-classification", model=model_emotion, tokenizer=tokenizer_emotion)
+    result = pipe(user_input)
+    emotion = result[0]['label']
+    
+    # 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", "link": "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"},
+            {"subject": "Dealing with Stress", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"},
+            {"subject": "Emotional Wellness Toolkit", "link": "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", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit"},
+            {"subject": "Stress Management Tips", "link": "https://www.health.harvard.edu/health-a-to-z"},
+            {"subject": "Dealing with Anger", "link": "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", "link": "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"},
+            {"subject": "Coping with Anxiety", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"},
+            {"subject": "Emotional Wellness Toolkit", "link": "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", "link": "https://www.nih.gov/health-information/emotional-wellness-toolkit"},
+            {"subject": "Dealing with Anxiety", "link": "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", "link": "https://www.health.harvard.edu/health-a-to-z"},
+            {"subject": "Coping Strategies", "link": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"}
+        ]
+        video_link = "Watch on YouTube: https://youtu.be/m1vaUGtyo-A"
+        
+    else:
+        emotion_msg = "Could not detect emotion."
+        resources = []
+        video_link = ""
+    
+    return emotion_msg, resources, video_link
+
+
 # Google Geocoding API setup to convert city name to latitude/longitude
 geocode_url = "https://maps.googleapis.com/maps/api/geocode/json"
 
@@ -152,116 +210,27 @@ def scrape_wellness_professionals(query, location):
     if response.status_code == 200:
         soup = BeautifulSoup(response.text, 'html.parser')
         
-        # Parse and extract wellness professionals from the HTML
-        wellness_data = []
-        results = soup.find_all("div", class_="BVG0Nb")  # Adjust class based on the actual HTML structure
-        for result in results:
-            name = result.get_text()
-            link = result.find("a")["href"] if result.find("a") else None
-            wellness_data.append([name, link])
+        # Find the results based on HTML structure
+        # Note: This is a simplistic example, Google search results structure may change
+        result_divs = soup.find_all("div", class_="BVG0Nb")
+        
+        results = []
+        for div in result_divs:
+            name = div.get_text()
+            link = div.find("a")["href"]
+            results.append({"name": name, "link": link})
         
-        return wellness_data
+        return results
     else:
-        return []
-
-# Updated function to 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 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
-
-    # Fallback to scraping if API is not available or fails
-    return scrape_wellness_professionals(query, location)
-
-# 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']
-    
-    # Provide suggestions based on the detected emotion
-    if emotion == 'joy':
-        return ("You're feeling happy! Keep up the great mood!", 
-                [("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")],
-                "Watch on YouTube: https://youtu.be/m1vaUGtyo-A")
-    elif emotion == 'anger':
-        return ("You're feeling angry. It's okay to feel this way. Let's try to calm down.", 
-                [("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")],
-                "Watch on YouTube: https://youtu.be/MIc299Flibs")
-    elif emotion == 'fear':
-        return ("You're feeling fearful. Take a moment to breathe and relax.", 
-                [("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")],
-                "Watch on YouTube: https://youtu.be/yGKKz185M5o")
-    elif emotion == 'sadness':
-        return ("You're feeling sad. It's okay to take a break.", 
-                [("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")],
-                "Watch on YouTube: https://youtu.be/-e-4Kx5px_I")
-    elif emotion == 'surprise':
-        return ("You're feeling surprised. It's okay to feel neutral!", 
-                [("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")],
-                "Watch on YouTube: https://youtu.be/m1vaUGtyo-A")
-    return ("Could not detect emotion.", [], "")
-
-# Gradio interface setup for user interaction
-def user_interface(message, location, history):
-    history, history = chat(message, history)
-    
-    # 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_msg, resources, video_link = detect_emotion(message)
-    
-    # Get wellness professionals
-    wellness_data = get_wellness_professionals(location, api_key)
-    
-    # Display wellness professionals in a table format
-    wellness_df = pd.DataFrame(wellness_data, columns=["Name", "Address", "Latitude", "Longitude"])
-    
-    return history, history, sentiment, emotion_msg, resources, video_link, wellness_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")
+        return None
 
-# Gradio interface definition
-demo = gr.Interface(
-    user_interface,
-    [gr.Textbox(label="Message"), location_input, "state"],
-    [chatbot, "state", "text", "text", "json", "text", "html"],
+# Initialize the chatbot interface
+iface = gr.Interface(
+    fn=chat,
+    inputs=["text", "state"],
+    outputs=["chatbot", "state"],
     allow_flagging="never",
-    title="Mental Health & Well-being Assistant"
+    live=True
 )
 
-# Launch Gradio interface
-if __name__ == "__main__":
-    demo.launch()
+iface.launch(share=True)