Update app.py

app.py

@@ -13,7 +13,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
 import torch
 import tensorflow as tf
 import tflearn
-import time
+from geopy.geocoders import Nominatim
 from bs4 import BeautifulSoup
 import re  # Added for regex operations
 
@@ -52,7 +52,13 @@ tokenizer, emotion_model = load_model()
 
 # Google Places API query function
 def get_places_data(query, location, radius=5000, api_key="GOOGLE_API_KEY"):
-    latitude, longitude = map(float, location.split(","))
+    # Use geopy to convert location name to coordinates
+    geolocator = Nominatim(user_agent="place_search")
+    location_obj = geolocator.geocode(location)
+    if location_obj is None:
+        return []
+
+    latitude, longitude = location_obj.latitude, location_obj.longitude
     params = {
         "query": query,
         "location": f"{latitude},{longitude}",
@@ -144,54 +150,26 @@ def scrape_website_for_contact_info(website):
     return phone_number, email
 
 # Main Gradio interface for emotion detection and chatbot
-def emotion_and_chatbot(user_input, history, query, location):
-    # Emotion Detection
-    emotion = detect_emotion(user_input)
-    sentiment = analyze_sentiment(user_input)
-    emotion_response = f"Emotion Detected: {emotion}. Sentiment: {sentiment}"
-    
-    # Provide suggestions based on emotion
-    suggestions = {
-        "joy": ["Relaxation Techniques", "Dealing with Stress", "Emotional Wellness Toolkit"],
-        "anger": ["Stress Management Tips", "Dealing with Anger", "Emotional Wellness Toolkit"],
-        "fear": ["Mindfulness Practices", "Coping with Anxiety", "Emotional Wellness Toolkit"],
-        "sadness": ["Dealing with Anxiety", "Emotional Wellness Toolkit"],
-        "surprise": ["Managing Stress", "Coping Strategies"]
-    }
-    
-    # Suggested articles and video links
-    if emotion in suggestions:
-        resources = suggestions[emotion]
-        links = {
-            "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",
-            "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",
-            "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",
-            "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"
-        }
-        
-        st.write("Useful Resources:")
-        for resource in resources:
-            st.markdown(f"[{resource}]({links[resource]})")
-
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/m1vaUGtyo-A)")
-    
-    # Search Places (for wellness or other queries)
-    places_data = get_places_data(query, location)
-    places_df = pd.DataFrame(places_data)
-    
-    # Tabular output for places
-    places_table = places_df[['name', 'vicinity', 'geometry']].head(10).to_html(classes='table table-bordered') if not places_df.empty else "No places found."
-    
-    # Generate Map
-    places_map = create_map(places_data) if places_data else "No places found."
-    
-    # Chatbot response
+def emotion_and_chatbot(user_input, history, query, location, emotion_button, search_button):
+    # Handle emotion detection
+    if emotion_button:
+        emotion = detect_emotion(user_input)
+        sentiment = analyze_sentiment(user_input)
+        emotion_response = f"Emotion Detected: {emotion}. Sentiment: {sentiment}"
+    else:
+        emotion_response = "Click on 'Detect Emotion' to analyze your input."
+
+    # Handle place search (therapists or wellness centers)
+    if search_button:
+        places_data = get_places_data(query, location)
+        places_df = pd.DataFrame(places_data)
+        places_table = places_df[['name', 'vicinity', 'geometry']].head(10).to_html(classes='table table-bordered') if not places_df.empty else "No places found."
+        places_map = create_map(places_data) if places_data else "No places found."
+    else:
+        places_table = "Click on 'Search for Therapists' to search."
+        places_map = ""
+
+    # Handle chatbot conversation
     history, _ = chatbot(user_input, history)
     
     return emotion_response, places_map, places_table, history, history
@@ -203,7 +181,9 @@ iface = gr.Interface(
         gr.Textbox(label="Enter your message", placeholder="How are you feeling?"),
         "state",  # Chat history
         gr.Textbox(label="Search Query (e.g. wellness)", placeholder="e.g. therapist"),
-        gr.Textbox(label="Location (latitude,longitude)", placeholder="e.g. 21.3,-157.8")
+        gr.Textbox(label="Location (e.g. Lahore, Hawaii, Allama Iqbal Town)", placeholder="e.g. Lahore, Allama Iqbal Town"),
+        gr.Button("Detect Emotion"),
+        gr.Button("Search for Therapists")
     ],
     outputs=[
         gr.Textbox(label="Emotion and Sentiment"),