Update app.py

app.py

@@ -6,7 +6,7 @@ from math import radians, cos, sin, sqrt, atan2
 from branca.element import MacroElement
 from jinja2 import Template
 
-# 1. Districts of Rwanda with coordinates
+# 1. Districts and Coordinates
 places = {
     "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),
@@ -20,7 +20,7 @@ places = {
     "Nyamasheke": (-2.4700, 29.3222), "Ngororero": (-1.8733, 29.5811)
 }
 
-# 2. Haversine function for distance (used in graph and heuristic)
+# 2. Distance calculator
 def haversine(coord1, coord2):
     R = 6371
     lat1, lon1 = coord1
@@ -31,7 +31,7 @@ def haversine(coord1, coord2):
     c = 2 * atan2(sqrt(a), sqrt(1 - a))
     return R * c
 
-# 3. Graph structure with real connections between neighboring districts
+# 3. District Connections
 edges = [
     ("Nyarugenge", "Gasabo"), ("Gasabo", "Kicukiro"), ("Kicukiro", "Bugesera"), ("Bugesera", "Rwamagana"),
     ("Rwamagana", "Kayonza"), ("Kayonza", "Kirehe"), ("Kirehe", "Ngoma"), ("Ngoma", "Gatsibo"),
@@ -44,17 +44,12 @@ edges = [
     ("Gicumbi", "Gasabo"), ("Bugesera", "Ngoma")
 ]
 
-# 4. Build the weighted graph
+# 4. Create Graph
 G = nx.Graph()
 for u, v in edges:
-    dist = haversine(places[u], places[v])
-    G.add_edge(u, v, weight=dist)
+    G.add_edge(u, v, weight=haversine(places[u], places[v]))
 
-# 5. A* heuristic function
-def heuristic(u, v):
-    return haversine(places[u], places[v])
-
-# 6. Folium animation class
+# 5. Animate Marker
 class AnimateMarker(MacroElement):
     _template = Template("""
         {% macro script(this, kwargs) %}
@@ -77,7 +72,7 @@ class AnimateMarker(MacroElement):
         self._name = "AnimateMarker"
         self.locations = locations
 
-# 7. Core route generation function
+# 6. Core route function
 def generate_map(start, end):
     if start == end:
         return "Hitamo aho utangiriye n’aho ugiye bitandukanye.", ""
@@ -86,7 +81,7 @@ def generate_map(start, end):
         return f"Nta nzira ibaho hagati ya {start} na {end}.", ""
 
     try:
-        path = nx.astar_path(G, start, end, heuristic=heuristic, weight='weight')
+        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)
@@ -100,7 +95,7 @@ def generate_map(start, end):
     except Exception as e:
         return f"Ntibishoboka kubona inzira: {str(e)}", ""
 
-# 8. Gradio Interface
+# 7. Gradio App
 iface = gr.Interface(
     fn=generate_map,
     inputs=[
@@ -111,7 +106,9 @@ iface = gr.Interface(
         gr.Textbox(label="Ubutumwa"),
         gr.HTML(label="Ikarita")
     ],
-    title="🗺️ VIATEUR AI Rwanda Smart Route Planner "
+    title="🗺️ VIATEUR AI - Rwanda Smart Route Planner",
+    theme="default"
 )
 
 iface.launch()
+