Update app.py

app.py

@@ -58,8 +58,10 @@ def held_karp_tsp(dist_matrix):
         else:
             break
 
-    tour.append(0)
-    return min_tour_cost, tour
+    if len(tour) == n + 1:
+        return min_tour_cost, tour
+    else:
+        return inf, []
 
 # Mendapatkan koordinat kota dari nama
 def get_coordinates(city_name):
@@ -75,7 +77,7 @@ def create_distance_matrix(coordinates):
     valid_coordinates = [c for c in coordinates if c is not None]
     n = len(valid_coordinates)
     if n < 2:
-        return [], []
+        return np.array([]), valid_coordinates
     dist_matrix = np.zeros((n, n))
     for i in range(n):
         for j in range(i + 1, n):
@@ -126,13 +128,16 @@ if st.button("Optimasi Rute"):
         min_cost, optimal_route = held_karp_tsp(dist_matrix)
         
         # Tampilkan hasil
-        st.write(f"Biaya total minimum: {min_cost:.2f} km")
-        st.write("Rute optimal:", " -> ".join([city_names[i] for i in optimal_route]))
-
-        # Buat peta
-        map_obj = folium.Map(location=valid_coordinates[0], zoom_start=5)
-        plot_route(map_obj, valid_coordinates, optimal_route)
-        
-        # Render map
-        st.markdown("### Rute Optimal")
-        st_folium = st.components.v1.html(folium.Map._repr_html_(map_obj), width=800, height=600)
+        if min_cost == float('inf'):
+            st.write("Tidak ada rute optimal yang dapat ditemukan.")
+        else:
+            st.write(f"Biaya total minimum: {min_cost:.2f} km")
+            st.write("Rute optimal:", " -> ".join([city_names[i] for i in optimal_route]))
+
+            # Buat peta
+            map_obj = folium.Map(location=valid_coordinates[0], zoom_start=5)
+            plot_route(map_obj, valid_coordinates, optimal_route)
+            
+            # Render map
+            st.markdown("### Rute Optimal")
+            st_folium = st.components.v1.html(folium.Map._repr_html_(map_obj), width=800, height=600)