Update app.py

app.py

@@ -7,6 +7,9 @@ import numpy as np
 
 # Algoritma Held-Karp untuk TSP
 def held_karp_tsp(dist_matrix):
+    if len(dist_matrix) < 2:
+        return 0, []
+
     n = len(dist_matrix)
     inf = float('inf')
     memo = {}
@@ -70,12 +73,14 @@ def get_coordinates(city_name):
 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 [], []
     dist_matrix = np.zeros((n, n))
     for i in range(n):
         for j in range(i + 1, n):
             dist_matrix[i][j] = geodesic(valid_coordinates[i], valid_coordinates[j]).kilometers
             dist_matrix[j][i] = dist_matrix[i][j]
-    return dist_matrix
+    return dist_matrix, valid_coordinates
 
 # Fungsi untuk menampilkan peta rute
 def plot_route(map_obj, coordinates, route):
@@ -98,19 +103,22 @@ st.title("Traveling Salesman Problem Solver")
 st.subheader("Pilih Kota")
 city_count = st.number_input("Jumlah kota", min_value=2, step=1)
 city_names = []
+city_coords = []
 for i in range(int(city_count)):
     city_name = st.text_input(f"Kota {i+1}", key=f"city_{i}")
-    city_names.append(city_name)
+    city_coords.append(get_coordinates(city_name))
+    if city_coords[-1] is None:
+        st.warning(f"Kota '{city_name}' tidak ditemukan. Harap periksa ejaan dan coba lagi.")
+    else:
+        city_names.append(city_name)
 
 if st.button("Optimasi Rute"):
-    coordinates = [get_coordinates(city) for city in city_names]
-    valid_coordinates = [c for c in coordinates if c is not None]
+    dist_matrix, valid_coordinates = create_distance_matrix(city_coords)
 
     if len(valid_coordinates) < 2:
-        st.error("Ada kota yang tidak ditemukan. Pastikan nama kota benar.")
+        st.error("Tidak cukup kota yang valid untuk dioptimalkan.")
     else:
-        # Buat distance matrix
-        dist_matrix = create_distance_matrix(valid_coordinates)
+        # Hitung rute optimal
         min_cost, optimal_route = held_karp_tsp(dist_matrix)
         
         # Tampilkan hasil