Update app.py

app.py

@@ -163,6 +163,23 @@ def get_lat_lon(location, api_key):
         print(f"Error fetching coordinates: {e}")
         return None, None
 
+# Function to fetch places data using Google Places API
+def get_places_data(query, location, radius, api_key):
+    places_url = "https://maps.googleapis.com/maps/api/place/textsearch/json"
+    params = {
+        "query": query,
+        "location": location,
+        "radius": radius,
+        "key": api_key
+    }
+    try:
+        response = requests.get(places_url, params=params)
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        print(f"Error fetching places data: {e}")
+        return None
+
 # 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"
@@ -226,53 +243,47 @@ def user_interface(message, location, history, api_key, words, labels, model):
     map_html = 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
-    suggestions_html = suggestions_df.to_html(escape=False)
+    suggestions_df = pd.DataFrame(resources, columns=["Resource", "Link"])
     
-    return history, history, sentiment, emotion, suggestions_html, map_html
-
-# Load data and model
-try:
-    data = load_intents("intents.json")
-except FileNotFoundError:
-    raise FileNotFoundError("Error: 'intents.json' file not found. Ensure it exists in the current directory.")
+    # Format the final output
+    output = f"**Chat Response:** {history[-1][1]}\n\n"
+    output += f"**Sentiment:** {sentiment}\n\n"
+    output += f"**Detected Emotion:** {emotion}\n\n"
+    output += "**Suggestions based on Emotion:**\n"
+    for resource in resources:
+        output += f"- [{resource[0]}]({resource[1]})\n"
+    output += f"\n**Video Suggestion:** {video_link}\n\n"
+    output += "**Wellness Professionals near you:**\n"
+    output += map_html
+    
+    return output, history
 
-try:
-    words, labels, training, output = load_preprocessed_data("data.pickle")
-except FileNotFoundError:
-    raise FileNotFoundError("Error: 'data.pickle' file not found. Ensure it exists and matches the model.")
+# Load intents and preprocessed data
+data = load_intents('intents.json')
+words, labels, training, output = load_preprocessed_data('data.pkl')
 
+# Build and load the model
 net = build_model(words, labels, training, output)
-try:
-    model = load_model("MentalHealthChatBotmodel.tflearn", net)
-except FileNotFoundError:
-    raise FileNotFoundError("Error: Trained model file 'MentalHealthChatBotmodel.tflearn' not found.")
-
-# 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")
+model = load_model('model.tflearn', net)
 
-# Gradio interface definition
-demo = gr.Interface(
-    fn=lambda message, location, history: user_interface(message, location, history, os.getenv("GOOGLE_API_KEY"), words, labels, model),
+# Gradio interface
+iface = gr.Interface(
+    fn=user_interface,
     inputs=[
-        gr.Textbox(label="Message"),
-        location_input,
-        "state",
+        gr.inputs.Textbox(label="Message"),
+        gr.inputs.Textbox(label="Location"),
+        gr.State([]),  # History
+        gr.State(os.getenv("GOOGLE_API_KEY")),  # API Key
+        gr.State(words),
+        gr.State(labels),
+        gr.State(model)
     ],
     outputs=[
-        chatbot, 
-        "state", 
-        gr.Textbox(label="Sentiment"), 
-        gr.Textbox(label="Emotion"), 
-        gr.HTML(label="Resources"), 
-        gr.HTML(label="Map")
+        gr.outputs.Markdown(label="Response"),
+        gr.outputs.Chatbot(label="Chat History")
     ],
-    allow_flagging="never",
-    title="Mental Health & Well-being Assistant"
+    title="Wellness Chatbot",
+    description="A chatbot to provide wellness support and locate professionals near you."
 )
 
-# Launch Gradio interface
-if __name__ == "__main__":
-    demo.launch(debug=True)
+iface.launch(debug=True)