Update app.py

app.py

@@ -1,73 +1,79 @@
 import gradio as gr
+import folium
+from folium.plugins import MarkerCluster
+import tempfile
+import webbrowser
 
 # =========================
-# DATABASE YA HOSPITALS MU RWANDA (with GPS + specialty)
+# DATABASE YA HOSPITALS MU RWANDA (GPS + specialty)
 # =========================
 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"]},
+     "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"]},
+     "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"]},
+     "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"]},
+     "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"]},
+     "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"]},
+     "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"]}
-    # Ongeraho hospitals zose wifuza, GPS na specialty
+     "contact": "", "email": "", "gps": [-1.641,29.893], "specialty": ["General","Oncology","Pediatrics"]}
 ]
 
 # =========================
-# FUNCTION YO GUSUBIZA IBIBAZO
+# FUNCTION YO GUSHIRAHO MAP
 # =========================
-def healthcare_ai(query, province="", district="", specialty=""):
-    """
-    Ushobora kwandika symptom, province, district cyangwa specialty.
-    AI izasubiza ibitaro bihuye n'aho uri cyangwa specialty wifuza.
-    """
+def healthcare_map(province="", district="", specialty=""):
+    # Filter hospitals
     results = hospitals_data
-
-    # Filter ku province niba yatanzwe
     if province:
         results = [h for h in results if h["province"].lower() == province.lower()]
-    # Filter ku district niba yatanzwe
     if district:
         results = [h for h in results if h["district"].lower() == district.lower()]
-    # Filter ku specialty niba yatanzwe
     if specialty:
         results = [h for h in results if specialty.lower() in [s.lower() for s in h["specialty"]]]
-    # Filter ku query keyword (hospital name / symptom keyword)
-    if query:
-        results = [h for h in results if query.lower() in h["name"].lower() or query.lower() in h["district"].lower() or query.lower() in h["sector"].lower()]
 
     if not results:
-        return "Ntibitaro bibonetse. Ongera ugerageze keyword, province, district cyangwa specialty."
+        return "Ntibitaro bibonetse. Ongera ugerageze keyword cyangwa filter."
+
+    # 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)
 
-    # Format response
-    response = "🏥 Ibitaro biboneka:\n\n"
     for h in results:
-        response += f"🏥 {h['name']}\n📍 Intara: {h['province']}\n📍 Akarere: {h['district']}\n📍 Umurenge: {h['sector']}\n📞 Contact: {h['contact'] if h['contact'] else 'N/A'}\n📧 Email: {h['email'] if h['email'] else 'N/A'}\n🌐 GPS: {h['gps']}\n⭐ Specialty: {', '.join(h['specialty'])}\n\n"
+        popup_text = 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'])}
+        """
+        folium.Marker(location=h["gps"], popup=popup_text).add_to(marker_cluster)
 
-    return response
+    # Save map to temporary HTML file
+    tmp_file = tempfile.NamedTemporaryFile(suffix=".html", delete=False)
+    rwanda_map.save(tmp_file.name)
+    return tmp_file.name
 
 # =========================
 # GRADIO INTERFACE
 # =========================
 iface = gr.Interface(
-    fn=healthcare_ai,
+    fn=healthcare_map,
     inputs=[
-        gr.Textbox(label="Andika symptom, hospital name, district cyangwa province", placeholder="Urugero: Kigali, Gahini, CHUK..."),
         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")
     ],
-    outputs="text",
-    title="Rwanda Healthcare AI – GPS + Specialty",
-    description="Shaka ibitaro mu Rwanda. Ushobora kwandika symptom, hospital name, province, district, cyangwa specialty, AI ikagusubiza ibitaro bihuye n'ibyo ukeneye."
+    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."
 )
 
 iface.launch()