Update app.py

app.py

@@ -6,8 +6,9 @@ from math import radians, cos, sin, sqrt, atan2
 from branca.element import MacroElement
 from jinja2 import Template
 
-# 1. Districts and Coordinates
+# 1. Districts and Coordinates (Harimo Kigali)
 places = {
+    "Kigali": (-1.9441, 30.0619),  # Kigali city center
     "Nyarugenge": (-1.9577, 30.0619), "Gasabo": (-1.9400, 30.0861), "Kicukiro": (-1.9781, 30.0597),
     "Burera": (-1.4800, 29.7300), "Gakenke": (-1.5700, 29.7561), "Rulindo": (-1.8333, 30.0833),
     "Musanze": (-1.5014, 29.6344), "Gicumbi": (-1.5794, 30.0542), "Nyagatare": (-1.3100, 30.3000),
@@ -20,9 +21,9 @@ places = {
     "Nyamasheke": (-2.4700, 29.3222), "Ngororero": (-1.8733, 29.5811)
 }
 
-# 2. Distance calculator
+# 2. Distance calculator (Haversine Formula)
 def haversine(coord1, coord2):
-    R = 6371
+    R = 6371  # Earth radius in km
     lat1, lon1 = coord1
     lat2, lon2 = coord2
     dlat = radians(lat2 - lat1)
@@ -31,7 +32,7 @@ def haversine(coord1, coord2):
     c = 2 * atan2(sqrt(a), sqrt(1 - a))
     return R * c
 
-# 3. District Connections
+# 3. Edges (connect Kigali and other districts)
 edges = [
     ("Nyarugenge", "Gasabo"), ("Gasabo", "Kicukiro"), ("Kicukiro", "Bugesera"), ("Bugesera", "Rwamagana"),
     ("Rwamagana", "Kayonza"), ("Kayonza", "Kirehe"), ("Kirehe", "Ngoma"), ("Ngoma", "Gatsibo"),
@@ -41,7 +42,10 @@ edges = [
     ("Ngororero", "Muhanga"), ("Muhanga", "Kamonyi"), ("Kamonyi", "Nyarugenge"), ("Muhanga", "Ruhango"),
     ("Ruhango", "Nyanza"), ("Nyanza", "Huye"), ("Huye", "Gisagara"), ("Gisagara", "Nyaruguru"),
     ("Nyaruguru", "Nyamagabe"), ("Nyamagabe", "Karongi"), ("Ngororero", "Ruhango"),
-    ("Gicumbi", "Gasabo"), ("Bugesera", "Ngoma")
+    ("Gicumbi", "Gasabo"), ("Bugesera", "Ngoma"),
+
+    # Kigali connections
+    ("Kigali", "Nyarugenge"), ("Kigali", "Gasabo"), ("Kigali", "Kicukiro")
 ]
 
 # 4. Create Graph
@@ -49,7 +53,7 @@ G = nx.Graph()
 for u, v in edges:
     G.add_edge(u, v, weight=haversine(places[u], places[v]))
 
-# 5. Animate Marker
+# 5. Animation class
 class AnimateMarker(MacroElement):
     _template = Template("""
         {% macro script(this, kwargs) %}
@@ -72,7 +76,7 @@ class AnimateMarker(MacroElement):
         self._name = "AnimateMarker"
         self.locations = locations
 
-# 6. Core route function
+# 6. Routing Function
 def generate_map(start, end):
     if start == end:
         return "Hitamo aho utangiriye n’aho ugiye bitandukanye.", ""
@@ -84,7 +88,7 @@ def generate_map(start, end):
         path = nx.astar_path(G, start, end, heuristic=lambda u, v: haversine(places[u], places[v]), weight='weight')
         coords = [places[p] for p in path]
 
-        m = Map(location=[-2.0, 30.0], zoom_start=8)
+        m = Map(location=[-1.9441, 30.0619], zoom_start=9)
         for name, coord in places.items():
             Marker(location=coord, popup=name).add_to(m)
 
@@ -95,7 +99,7 @@ def generate_map(start, end):
     except Exception as e:
         return f"Ntibishoboka kubona inzira: {str(e)}", ""
 
-# 7. Gradio App
+# 7. Gradio Interface
 iface = gr.Interface(
     fn=generate_map,
     inputs=[
@@ -111,4 +115,3 @@ iface = gr.Interface(
 )
 
 iface.launch()
-