Update app.py

app.py

@@ -1,13 +1,73 @@
 import gradio as gr
+import pandas as pd
+import numpy as np
+from ortools.constraint_solver import pywrapcp, routing_enums_pb2
 
-def test(name):
-    return f"Hello {name}, your Gradio Space is working!"
+def optimize_route(address_text):
 
+    # Parse addresses
+    addresses = [a.strip() for a in address_text.split("\n") if a.strip()]
+    if len(addresses) < 2:
+        return "❌ Please enter at least two locations.", None
+
+    # Generate fake symmetric distance matrix (Phase 1 demo)
+    n = len(addresses)
+    np.random.seed(42)
+    distance_matrix = np.random.randint(5, 50, size=(n, n))
+    np.fill_diagonal(distance_matrix, 0)
+
+    # OR-Tools Manager
+    manager = pywrapcp.RoutingIndexManager(n, 1, 0)
+    routing = pywrapcp.RoutingModel(manager)
+
+    def distance_callback(from_idx, to_idx):
+        return distance_matrix[manager.IndexToNode(from_idx)][manager.IndexToNode(to_idx)]
+
+    callback_idx = routing.RegisterTransitCallback(distance_callback)
+    routing.SetArcCostEvaluatorOfAllVehicles(callback_idx)
+
+    search_params = pywrapcp.DefaultRoutingSearchParameters()
+    search_params.first_solution_strategy = (
+        routing_enums_pb2.FirstSolutionStrategy.PATH_CHEAPEST_ARC
+    )
+
+    solution = routing.SolveWithParameters(search_params)
+    if not solution:
+        return "❌ Route optimization failed.", None
+
+    # Extract route
+    index = routing.Start(0)
+    ordered_stops = []
+    while not routing.IsEnd(index):
+        ordered_stops.append(manager.IndexToNode(index))
+        index = solution.Value(routing.NextVar(index))
+
+    ordered_addresses = [addresses[i] for i in ordered_stops]
+
+    # Build readable text
+    route_text = "🚚 **Optimized Route:**\n\n"
+    for i, loc in enumerate(ordered_addresses):
+        route_text += f"**{i+1}. {loc}**\n"
+
+    df = pd.DataFrame(distance_matrix, columns=addresses, index=addresses)
+
+    return route_text, df
+
+
+# Gradio interface
 interface = gr.Interface(
-    fn=test,
-    inputs="text",
-    outputs="text",
-    title="Procelevate Route Planner Test"
+    fn=optimize_route,
+    inputs=gr.Textbox(
+        lines=8,
+        label="Enter addresses (one per line):",
+        placeholder="Example:\nBangalore Airport\nMG Road\nElectronic City\nWhitefield",
+    ),
+    outputs=[
+        gr.Markdown(label="Optimized Route"),
+        gr.Dataframe(label="Distance Matrix (Demo Mode)")
+    ],
+    title="Procelevate AI Route Planner",
+    description="AI-powered optimization for logistics & 3PL route planning.",
 )
 
 interface.launch()