Update app.py

app.py

@@ -7,86 +7,64 @@ import os
 # Initialize Google Maps API client securely
 gmaps = googlemaps.Client(key=os.getenv('GOOGLE_API_KEY'))  # Fetch API key from Hugging Face secrets
 
-# Function to fetch directions
-def get_directions_and_coords(source, destination):
-    now = datetime.now()
-    directions_info = gmaps.directions(source, destination, mode='driving', departure_time=now)
-    if directions_info:
-        steps = directions_info[0]['legs'][0]['steps']
-        coords = [(step['start_location']['lat'], step['start_location']['lng']) for step in steps]
-        return steps, coords
-    else:
-        return None, None
-
-# Function to render map with directions
-def render_folium_map(coords):
-    m = folium.Map(location=[coords[0][0], coords[0][1]], zoom_start=13)
-    folium.PolyLine(coords, color="blue", weight=2.5, opacity=1).add_to(m)
-    return m
-
-# Function to search nearby medical centers using Google Places API
-def search_medical_centers(query, location, radius=10000):
-    places_result = gmaps.places_nearby(location, radius=radius, type='hospital', keyword=query)
+# Function to search for health professionals using Google Places API
+def search_health_professionals(query, location, radius=10000):
+    # Use the query to search for places like doctors, psychiatrists, psychologists, etc., within the specified radius.
+    places_result = gmaps.places_nearby(location, radius=radius, type='doctor', keyword=query)
     return places_result.get('results', [])
 
 # Function to get directions and display on Gradio UI
-def get_route_and_medical_centers(current_location, destination_location, health_professional_query):
+def get_health_professionals_and_map(current_location, health_professional_query):
     route_info = ""
     m = None  # Default to None
     
-    # Fetch the directions for the route
     try:
-        steps, coords = get_directions_and_coords(current_location, destination_location)
+        # Geocode the current location (i.e., convert it to latitude and longitude)
+        geocode_result = gmaps.geocode(current_location)
+        if not geocode_result:
+            route_info = "Could not retrieve location coordinates. Please enter a valid location."
+            return route_info, m
+        
+        location_coords = geocode_result[0]['geometry']['location']
+        lat, lon = location_coords['lat'], location_coords['lng']
+        
+        # Search for health professionals based on the user's query near the current location
+        health_professionals = search_health_professionals(health_professional_query, (lat, lon))
+        
+        if health_professionals:
+            route_info = "Health professionals found:\n"
+            m = folium.Map(location=[lat, lon], zoom_start=12)
+            for professional in health_professionals:
+                name = professional['name']
+                vicinity = professional.get('vicinity', 'N/A')
+                rating = professional.get('rating', 'N/A')
+                folium.Marker([professional['geometry']['location']['lat'], professional['geometry']['location']['lng']], 
+                              popup=f"{name}\n{vicinity}\nRating: {rating}").add_to(m)
+                route_info += f"- {name} ({rating} stars): {vicinity}\n"
+        else:
+            route_info = "No health professionals found matching your query."
+            m = folium.Map(location=[lat, lon], zoom_start=12)  # Default map if no professionals are found
+        
     except Exception as e:
-        return f"Error fetching directions: {str(e)}", None
-    
-    if steps and coords:
-        route_info = [f"{i+1}. {step['html_instructions']}" for i, step in enumerate(steps)]
-        m = render_folium_map(coords)
-    else:
-        route_info = "No available routes."
-        m = folium.Map(location=[20, 0], zoom_start=2)  # Default map if no route
-    
-    # Fetch nearby medical centers if the location input is provided
-    if current_location and health_professional_query:
-        try:
-            geocode_result = gmaps.geocode(current_location)
-            if geocode_result:
-                location_coords = geocode_result[0]['geometry']['location']
-                lat, lon = location_coords['lat'], location_coords['lng']
-                
-                medical_centers = search_medical_centers(health_professional_query, (lat, lon))
-                
-                if medical_centers:
-                    for center in medical_centers:
-                        name = center['name']
-                        vicinity = center.get('vicinity', 'N/A')
-                        rating = center.get('rating', 'N/A')
-                        folium.Marker([center['geometry']['location']['lat'], center['geometry']['location']['lng']], 
-                                      popup=f"{name}\n{vicinity}\nRating: {rating}").add_to(m)
-                else:
-                    route_info += "\nNo medical centers found matching your query."
-            else:
-                route_info += "\nCould not retrieve location coordinates."
-        except Exception as e:
-            route_info += f"\nError fetching medical centers: {str(e)}"
+        route_info = f"Error: {str(e)}"
+        m = folium.Map(location=[20, 0], zoom_start=2)  # Default map if any error occurs
     
     # Return both the route info and the map
-    return "\n".join(route_info), m._repr_html_()
+    return route_info, m._repr_html_()
 
 # Gradio UI components (Updated for Gradio v3.x)
-current_location_input = gr.Textbox(value="Honolulu, HI", label="Current Location (Plain Text)")
-destination_location_input = gr.Textbox(value="Maui, HI", label="Destination Location (Plain Text)")
-health_professional_query_input = gr.Textbox(value="Hawaii Medical Center", label="Health Professional Query (Plain Text)")
+current_location_input = gr.Textbox(value="Honolulu, HI", label="Current Location (Hawaii)")
+
+health_professional_query_input = gr.Textbox(value="doctor", label="Health Professional Query (e.g., doctor, psychiatrist, psychologist)")
 
 # Output components (Updated for Gradio v3.x)
-route_info_output = gr.Textbox(label="Driving Directions")
-map_output = gr.HTML(label="Map with Route and Health Professionals")
+route_info_output = gr.Textbox(label="Health Professionals Information")
+map_output = gr.HTML(label="Map with Health Professionals")
 
 # Create Gradio interface
 iface = gr.Interface(
-    fn=get_route_and_medical_centers,  # Function to call
-    inputs=[current_location_input, destination_location_input, health_professional_query_input],  # Inputs
+    fn=get_health_professionals_and_map,  # Function to call
+    inputs=[current_location_input, health_professional_query_input],  # Inputs
     outputs=[route_info_output, map_output],  # Outputs
     live=False  # Disable live updates, use button click to trigger the function
 )