update

app.py

@@ -104,9 +104,91 @@ def demo_plot(city, facility):
 
 
 def solver_plot(data_npy, boost=False):
-    # txt to numpy
+    data = data_npy.split('\n')
+    n = len(data)
+    p = int((len(data[0].split(' '))-2) / 2)
+    
+    positions = []
+    demands = []
+    actual_facilities = []
+    for row in data:
+        row = row.split(' ')
+        row = [x for x in row if len(x)]
+        print(row)
+        
+        positions.append([float(row[0]), float(row[1])])
+        
+        demand = []
+        for i in range(2, 2+p):
+            demand.append(float(row[i]))
+        demands.append(demand)
+        
+        actual_facility = []
+        for i in range(2+p, 2+2*p):
+            actual_facility.append(bool(int(row[i])))
+        actual_facilities.append(actual_facility)
+    positions = np.array(positions)
+    demands = np.array(demands)
+    actual_facilities = np.array(actual_facilities)
+    solution_facilities =  ~actual_facilities
+    print(actual_facilities)
+        
+    actual_fig = go.Figure()
+    solution_fig = go.Figure()
+    for i in range(p):
+        actual_fig.add_trace(go.Scattermapbox(
+            lat=positions[actual_facilities[:, i]][:, 0],
+            lon=positions[actual_facilities[:, i]][:, 1],
+            mode='markers',
+            marker=go.scattermapbox.Marker(
+                size=10,
+                color=px.colors.qualitative.Plotly[i]
+            ),
+            name=f'Facility {i+1}'
+        ))
+        solution_fig.add_trace(go.Scattermapbox(
+            lat=positions[solution_facilities[:, i]][:, 0],
+            lon=positions[solution_facilities[:, i]][:, 1],
+            mode='markers',
+            marker=go.scattermapbox.Marker(
+                size=10,
+                color=px.colors.qualitative.Plotly[i]
+            ),
+            name=f'Facility {i+1}'
+        ))
+    
+        actual_fig.update_layout(
+        mapbox=dict(
+            style='carto-positron',
+            center=dict(lat=np.mean(positions[actual_facilities[:, i]][:, 0]), \
+                lon=np.mean(positions[actual_facilities[:, i]][:, 1])),
+            zoom=11.0
+        ),
+        margin=dict(l=0, r=0, b=0, t=0),)
+
+        solution_fig.update_layout(
+        mapbox=dict(
+            style='carto-positron',
+            center=dict(lat=np.mean(positions[solution_facilities[:, i]][:, 0]), \
+                lon=np.mean(positions[solution_facilities[:, i]][:, 1])),
+            zoom=11.0
+        ),
+        margin=dict(l=0, r=0, b=0, t=0),)
+        # show legend
+        actual_fig.update_layout(showlegend=True)
+        solution_fig.update_layout(showlegend=True)
+    
+    positions = np.deg2rad(positions)
+    dist = pairwise_distances(positions, metric='haversine') * 6371
+    actual_ac = 0
+    solution_ac = 0
+    for i in range(p):
+        ac_matrix = dist * demands[:, i][:, None]
+        actual_ac += ac_matrix[:, actual_facilities[:, i]].min(axis=-1).sum()
+        solution_ac += ac_matrix[:, solution_facilities[:, i]].min(axis=-1).sum()
+        
+    return actual_fig, solution_fig, actual_ac, solution_ac
     
-    data = np.loadtxt(data_npy)
 
 
 
@@ -166,4 +248,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(server_port=1111)
+    demo.launch()