Update app.py

app.py

@@ -10,8 +10,6 @@ from nltk.tokenize import word_tokenize
 from nltk.stem.lancaster import LancasterStemmer
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import requests
-from bs4 import BeautifulSoup
-import pandas as pd
 import geocoder  # Use geocoder to get latitude/longitude from city
 
 # Ensure necessary NLTK resources are downloaded
@@ -108,13 +106,11 @@ def chat(message, history, state):
 tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
 sentiment_model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
 
-# Function for sentiment analysis
 # Function for sentiment analysis
 def analyze_sentiment(text, state):
-    # Initialize state if it's None
     if state is None:
         state = {'step': 1}
-    
+        
     inputs = tokenizer(text, return_tensors="pt")
     with torch.no_grad():
         outputs = sentiment_model(**inputs)
@@ -194,56 +190,60 @@ def provide_suggestions(emotion):
             'videos': "[Watch Stress Relief Video](https://youtu.be/m1vaUGtyo-A)"
         }
     }
-    return resources.get(emotion, {'message': "Stay calm. 🙂", 'articles': [], 'videos': []})
+    
+    # Ensure we return a message even if no articles/videos are found
+    return resources.get(emotion, {'message': "Stay calm. 🙂", 'articles': ["[General Wellbeing Tips](https://www.helpguide.org)"], 'videos': []})
 
-# Function to find wellness professionals
-def find_wellness_professionals(location, state):
-    # Geocode the location to get latitude and longitude
-    g = geocoder.osm(location)  # Using OpenStreetMap's geocoding service
+# Function to fetch wellness professionals based on location
+def get_wellness_professionals(location):
+    # Use Geocoder to get latitude/longitude from city
+    g = geocoder.osm(location)
     if g.ok:
-        location_coords = f"{g.lat},{g.lng}"
-    else:
-        return "Sorry, could not retrieve coordinates for the location. Please try again.", state
-    
-    query = "therapist OR counselor OR mental health professional OR marriage and family therapist OR psychotherapist OR psychiatrist OR psychologist in " + location
-    api_key = "GOOGLE_API_KEY"  # Replace with your own API key
-    radius = 50000  # 50 km radius
-    
-    google_places_data = get_all_places(query, location_coords, radius, api_key)
-    if google_places_data:
-        response = "Wellness professionals near you:\n"
-        for place in google_places_data:
-            response += f"- {place['name']} at {place['formatted_address']}\n"
+        latitude, longitude = g.latlng
+        google_api_url = f"https://maps.googleapis.com/maps/api/place/nearbysearch/json?location={latitude},{longitude}&radius=5000&type=health&key=YOUR_GOOGLE_API_KEY"
+        
+        response = requests.get(google_api_url)
+        data = response.json()
+        
+        professionals = []
+        if 'results' in data:
+            for place in data['results']:
+                name = place['name']
+                address = place.get('vicinity', 'No address available')
+                url = place.get('website', '#')
+                professionals.append(f"**{name}** - {address} - [Visit Website]({url})")
+        
+        if not professionals:
+            professionals.append("No wellness professionals found nearby.")
+        
+        return "\n".join(professionals)
     else:
-        response = "Sorry, no wellness professionals found in your area. Please try another location."
-
-    return response, state
+        return "Couldn't fetch your location. Please make sure you entered a valid location."
 
-# Call Google Places API
-def get_all_places(query, location, radius, api_key):
-    search_url = f"https://maps.googleapis.com/maps/api/place/textsearch/json?query={query}&location={location}&radius={radius}&key={api_key}"
-    response = requests.get(search_url).json()
-    
-    if 'results' in response:
-        return response['results']
-    return []
+# Function to ask for location and provide wellness professionals
+def search_wellness_professionals(location, state):
+    professionals = get_wellness_professionals(location)
+    state['step'] = 5  # Move to the next step
+    return professionals, state
 
-# Gradio UI components
-# Function to create the UI with state initialization
+# Create the UI with location input for wellness professionals
 def create_ui():
     with gr.Blocks() as demo:
-        state = gr.State({'step': 1})  # Initialize state with a default value
+        state = gr.State({'step': 1})
         chatbot = gr.Chatbot(elem_id="chatbot", label="Mental Health Chatbot")
         message_input = gr.Textbox(placeholder="Ask me something...", label="Enter your message")
         sentiment_output = gr.Textbox(placeholder="Sentiment result", label="Sentiment")
         emotion_output = gr.Textbox(placeholder="Detected emotion", label="Emotion")
-        wellness_output = gr.Textbox(placeholder="Wellness professional list", label="Wellness Professionals")
-        
+        wellness_output = gr.Textbox(placeholder="Wellness professionals nearby", label="Wellness Professionals")
+        location_input = gr.Textbox(placeholder="Enter your city for wellness professionals", label="Location")
+
         message_input.submit(chat, [message_input, chatbot, state], [chatbot, chatbot, state])
         message_input.submit(analyze_sentiment, [message_input, state], [sentiment_output, state])
         sentiment_output.submit(detect_emotion, [sentiment_output, state], [emotion_output, wellness_output, state])
-        return demo
+        location_input.submit(search_wellness_professionals, [location_input, state], [wellness_output, state])
+
+    return demo
 
-# Launch the Gradio interface
+# Launch Gradio interface
 demo = create_ui()
-demo.launch(debug=True)
+demo.launch(debug=True)