Update app.py

app.py

@@ -15,9 +15,7 @@ import torch
 import pandas as pd
 import time
 from bs4 import BeautifulSoup
-from selenium import webdriver
-from selenium.webdriver.chrome.options import Options
-import chromedriver_autoinstaller
+import re  # Added for regex operations
 import os
 
 # Google Places API endpoint
@@ -46,7 +44,6 @@ model = tflearn.DNN(net)
 model.load("MentalHealthChatBotmodel.tflearn")
 
 # Emotion and sentiment analysis model
-@st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
     model = AutoModelForSequenceClassification.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
@@ -65,7 +62,7 @@ def get_places_data(query, location, radius=5000, api_key="GOOGLE_API_KEY"):
     response = requests.get(url, params=params)
     if response.status_code == 200:
         data = response.json()
-        return data['results']
+        return data.get('results', [])
     else:
         return []
 
@@ -77,7 +74,7 @@ def create_map(locations):
         lat = loc['geometry']['location']['lat']
         lng = loc['geometry']['location']['lng']
         folium.Marker([lat, lng], popup=name).add_to(m)
-    return m
+    return m._repr_html_()  # Return HTML representation
 
 # Sentiment Analysis function
 def analyze_sentiment(user_input):
@@ -134,7 +131,7 @@ def scrape_website_for_contact_info(website):
     try:
         response = requests.get(website, timeout=5)
         soup = BeautifulSoup(response.content, 'html.parser')
-        phone_match = re.search(r'\(?\+?[0-9]*\)?[0-9_\- \(\)]*', soup.get_text())
+        phone_match = re.search(r'$$?\+?[0-9]*$$?[0-9_\- $$$$]*', soup.get_text())
         if phone_match:
             phone_number = phone_match.group()
         email_match = re.search(r'[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}', soup.get_text())
@@ -144,17 +141,6 @@ def scrape_website_for_contact_info(website):
         print(f"Error scraping website {website}: {e}")
     return phone_number, email
 
-# Main Gradio interface for location search and map generation
-def search_and_display_location(query, location):
-    query = query or "wellness"
-    location = location or "21.3,-157.8"  # Default location (Hawaii)
-    places_data = get_places_data(query, location)
-    if places_data:
-        map_object = create_map(places_data)
-        return map_object
-    else:
-        return "No results found for the given location."
-
 # Main Gradio interface for emotion detection and chatbot
 def emotion_and_chatbot(user_input, history, query, location):
     # Emotion Detection
@@ -165,7 +151,7 @@ def emotion_and_chatbot(user_input, history, query, location):
     # Search Places (for wellness or other queries)
     places_data = get_places_data(query, location)
     places_map = create_map(places_data) if places_data else "No places found."
-
+    
     # Chatbot response
     history, _ = chatbot(user_input, history)
     
@@ -192,4 +178,4 @@ iface = gr.Interface(
 
 # Launch Gradio app
 if __name__ == "__main__":
-    iface.launch(debug=True)
+    iface.launch(debug=True)