Update app.py

app.py

@@ -1,79 +1,104 @@
 import gradio as gr
 import folium
 from folium.plugins import MarkerCluster
+from geopy.distance import geodesic
+import spacy
 import tempfile
-import webbrowser
+
+# Load NLP model
+nlp = spacy.load("en_core_web_sm")
 
 # =========================
-# DATABASE YA HOSPITALS MU RWANDA (GPS + specialty)
+# DATABASE YA HOSPITALS
 # =========================
 hospitals_data = [
-    {"name": "King Faisal Hospital", "province": "Kigali City", "district": "Gasabo", "sector": "Kacyiru",
-     "contact": "+250788123200", "email": "info@kingfaisalhosp.rw", "gps": [-1.949,30.089], "specialty": ["General","Maternity","Surgery"]},
-    {"name": "CHUK", "province": "Kigali City", "district": "Nyarugenge", "sector": "Nyarugenge",
-     "contact": "", "email": "", "gps": [-1.950,30.058], "specialty": ["General","Pediatrics","Emergency"]},
-    {"name": "Gahini District Hospital", "province": "Eastern Province", "district": "Kayonza", "sector": "Gahini",
-     "contact": "0781403761", "email": "gahini.hospital@moh.gov.rw", "gps": [-2.019,30.576], "specialty": ["General","Orthopedic"]},
-    {"name": "Rwinkwavu District Hospital", "province": "Eastern Province", "district": "Kayonza", "sector": "Rwinkwavu",
-     "contact": "0788358810", "email": "rwinkwavu.hospital@moh.gov.rw", "gps": [-2.151,30.565], "specialty": ["General","Maternity"]},
-    {"name": "CHUB (Butare University Teaching Hospital)", "province": "Southern Province", "district": "Huye", "sector": "Ngoma",
-     "contact": "", "email": "", "gps": [-2.590,29.739], "specialty": ["General","Pediatrics","Surgery"]},
-    {"name": "Shyira Hospital", "province": "Western Province", "district": "Nyamasheke", "sector": "Bushenge",
-     "contact": "", "email": "", "gps": [-2.482,29.155], "specialty": ["General","Maternity"]},
-    {"name": "Butaro Hospital", "province": "Northern Province", "district": "Burera", "sector": "Butaro",
-     "contact": "", "email": "", "gps": [-1.641,29.893], "specialty": ["General","Oncology","Pediatrics"]}
+    {"name": "King Faisal Hospital", "province": "Kigali City", "district": "Gasabo",
+     "gps": (-1.949,30.089), "specialty": ["general","maternity","surgery"]},
+
+    {"name": "CHUK", "province": "Kigali City", "district": "Nyarugenge",
+     "gps": (-1.950,30.058), "specialty": ["general","pediatrics","emergency"]},
+
+    {"name": "Butaro Hospital", "province": "Northern Province", "district": "Burera",
+     "gps": (-1.641,29.893), "specialty": ["general","oncology","pediatrics"]},
 ]
 
 # =========================
-# FUNCTION YO GUSHIRAHO MAP
+# NLP FUNCTION
+# =========================
+def extract_specialty(text):
+    text = text.lower()
+    for sp in ["maternity","pediatrics","oncology","surgery","emergency","general"]:
+        if sp in text:
+            return sp
+    return ""
+
+# =========================
+# MAIN AI FUNCTION
 # =========================
-def healthcare_map(province="", district="", specialty=""):
-    # Filter hospitals
-    results = hospitals_data
-    if province:
-        results = [h for h in results if h["province"].lower() == province.lower()]
-    if district:
-        results = [h for h in results if h["district"].lower() == district.lower()]
-    if specialty:
-        results = [h for h in results if specialty.lower() in [s.lower() for s in h["specialty"]]]
+def ai_healthcare_assistant(user_text, user_lat, user_lon):
+    specialty = extract_specialty(user_text)
+
+    results = []
+    user_location = (user_lat, user_lon)
+
+    for h in hospitals_data:
+        if specialty and specialty not in h["specialty"]:
+            continue
+
+        distance = geodesic(user_location, h["gps"]).km
+        h_copy = h.copy()
+        h_copy["distance"] = round(distance, 2)
+        results.append(h_copy)
 
     if not results:
-        return "Ntibitaro bibonetse. Ongera ugerageze keyword cyangwa filter."
+        return "Nta bitaro bibonetse", None
+
+    results = sorted(results, key=lambda x: x["distance"])
 
-    # Create folium map centered in Rwanda
-    rwanda_map = folium.Map(location=[-1.940, 29.873], zoom_start=7)
-    marker_cluster = MarkerCluster().add_to(rwanda_map)
+    # Create map
+    m = folium.Map(location=[user_lat, user_lon], zoom_start=8)
+    cluster = MarkerCluster().add_to(m)
+
+    folium.Marker(
+        location=[user_lat, user_lon],
+        popup="YOU ARE HERE",
+        icon=folium.Icon(color="blue")
+    ).add_to(m)
 
     for h in results:
-        popup_text = f"""
+        popup = f"""
         <b>{h['name']}</b><br>
-        Intara: {h['province']}<br>
-        Akarere: {h['district']}<br>
-        Umurenge: {h['sector']}<br>
-        Contact: {h['contact'] if h['contact'] else 'N/A'}<br>
-        Email: {h['email'] if h['email'] else 'N/A'}<br>
-        Specialty: {', '.join(h['specialty'])}
+        District: {h['district']}<br>
+        Specialty: {', '.join(h['specialty'])}<br>
+        Distance: {h['distance']} km
         """
-        folium.Marker(location=h["gps"], popup=popup_text).add_to(marker_cluster)
+        folium.Marker(location=h["gps"], popup=popup).add_to(cluster)
 
-    # Save map to temporary HTML file
-    tmp_file = tempfile.NamedTemporaryFile(suffix=".html", delete=False)
-    rwanda_map.save(tmp_file.name)
-    return tmp_file.name
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".html")
+    m.save(tmp.name)
+
+    text_response = "🏥 Ibitaro byegereye aho uri:\n"
+    for h in results:
+        text_response += f"- {h['name']} ({h['distance']} km)\n"
+
+    return text_response, tmp.name
 
 # =========================
 # GRADIO INTERFACE
 # =========================
 iface = gr.Interface(
-    fn=healthcare_map,
+    fn=ai_healthcare_assistant,
     inputs=[
-        gr.Dropdown(label="Province (optional)", choices=[""] + [h["province"] for h in hospitals_data], type="value"),
-        gr.Textbox(label="District (optional)", placeholder="Akarere niba ushaka"),
-        gr.Dropdown(label="Specialty (optional)", choices=[""] + ["General","Maternity","Pediatrics","Surgery","Orthopedic","Psychiatric","Oncology"], type="value")
+        gr.Textbox(label="Vuga ikibazo cyawe (ex: I need pediatric hospital near me)"),
+        gr.Number(label="Latitude (aho uri)"),
+        gr.Number(label="Longitude (aho uri)")
+    ],
+    outputs=[
+        gr.Textbox(label="AI Recommendation"),
+        gr.File(label="Hospital Map")
     ],
-    outputs=gr.File(label="Interactive Map of Hospitals"),
-    title="Rwanda Healthcare AI – Interactive Map",
-    description="Shaka ibitaro mu Rwanda, urabona hospitals kuri interactive map. Ushobora filter province, district, specialty."
+    title="AI Healthcare Assistant – Rwanda (Advanced)",
+    description="AI ikumva icyo ushaka, ikamenya ibitaro byegereye aho uri hakurikijwe specialty."
 )
 
 iface.launch()