Update wisata_rekomendasi function to include 'gambar' column in DataFrame outputs

app.py

@@ -83,21 +83,21 @@ def sort_by_nearest_location(df, user_lat, user_lon):
 # --- Fungsi Utama Gradio ---
 def wisata_rekomendasi(deskripsi, lokasi):
     if df.empty:
-        return "Data tidak tersedia.", pd.DataFrame([["Data tidak tersedia", "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan"])
+        return "Data tidak tersedia.", pd.DataFrame([["Data tidak tersedia", "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan","gambar"])
 
     keywords = extract_keywords(deskripsi)
     if "Error:" in str(keywords):
-        return f"Kata kunci gagal diambil: {keywords[0]}", pd.DataFrame([[keywords[0], "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan"])
+        return f"Kata kunci gagal diambil: {keywords[0]}", pd.DataFrame([[keywords[0], "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan","gambar"])
     
     user_description_joined = " ".join(keywords)
     lat, lon = get_coordinates_from_location(lokasi)
     if lat is None or lon is None:
-        return "Lokasi tidak ditemukan.", pd.DataFrame([["Lokasi tidak ditemukan", "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan"])
+        return "Lokasi tidak ditemukan.", pd.DataFrame([["Lokasi tidak ditemukan", "", ""]], columns=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan","gambar"])
 
     top_place = prepare_and_recommend(df.copy(), user_description_joined)
     sorted_place = sort_by_nearest_location(top_place, lat, lon)
 
-    return f"Kata kunci: {', '.join(keywords)}", sorted_place[["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan"]]
+    return f"Kata kunci: {', '.join(keywords)}", sorted_place[["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan","gambar"]]
 
 # --- UI Gradio ---
 demo = gr.Interface(
@@ -108,7 +108,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Kata Kunci yang Diekstrak"),
-        gr.Dataframe(headers=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan"], label="Rekomendasi Tempat Wisata")
+        gr.Dataframe(headers=["id","nama","alamat", "distance_km", "deskripsi","harga","rating","total_ulasan","gambar"], label="Rekomendasi Tempat Wisata")
     ],
     title="Sistem Rekomendasi Wisata",
     description="Masukkan deskripsi dan lokasi, lalu dapatkan rekomendasi tempat wisata terdekat"