Add : Recommendation system

app.py

@@ -1,7 +1,109 @@
 import gradio as gr
+import re
+import ast
+import pandas as pd
+from geopy.geocoders import Nominatim
+from geopy.distance import geodesic
+from geopy.exc import GeocoderTimedOut
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
+from google import genai
+from dotenv import load_dotenv
+import os
 
-def greet(name):
-    return "Hello " + name + "!!"
+# --- Load API Key dari .env ---
+load_dotenv()
+API_KEY = os.getenv("GEMINI_API_KEY")
+genai.configure(api_key=API_KEY)
+
+# --- Load Data dari cleaned_data.csv ---
+try:
+    df = pd.read_csv("cleaned_data.csv")
+except FileNotFoundError:
+    print("File cleaned_data.csv tidak ditemukan.")
+    df = pd.DataFrame(columns=['nama', 'description', 'latitude', 'longitude'])
+
+# --- Fungsi Ekstraksi Kata Kunci ---
+def extract_keywords(user_input):
+    prompt = f"""
+    Ekstrak 3–7 kata kunci penting dari deskripsi wisata berikut:
+    "{user_input}"
+    Tulis langsung sebagai list Python tanpa variabel apapun.
+    """
+    try:
+        response = genai.GenerativeModel("gemini-1.5-flash").generate_content(prompt)
+        matches = re.findall(r'\[.*?\]', response.text)
+        if matches:
+            return ast.literal_eval(matches[0])
+        else:
+            return []
+    except Exception as e:
+        return [f"Error: {e}"]
+
+# --- Fungsi Lokasi ---
+def get_coordinates_from_location(location_name):
+    try:
+        geolocator = Nominatim(user_agent="geoapi")
+        location = geolocator.geocode(location_name, timeout=10)
+        return (location.latitude, location.longitude) if location else (None, None)
+    except GeocoderTimedOut:
+        return (None, None)
+
+def get_location_name_from_coordinates(lat, lon):
+    try:
+        geolocator = Nominatim(user_agent="geoapi")
+        location = geolocator.reverse((lat, lon), timeout=10)
+        return location.address if location else "Tidak ditemukan"
+    except GeocoderTimedOut:
+        return "Tidak ditemukan"
+
+# --- Rekomendasi Tempat Wisata ---
+def prepare_and_recommend(df, user_description):
+    tfidf = TfidfVectorizer()
+    tfidf_matrix = tfidf.fit_transform(df['description'].astype(str).tolist() + [user_description])
+    similarity = cosine_similarity(tfidf_matrix[-1], tfidf_matrix[:-1]).flatten()
+    df['similarity'] = similarity
+    return df.sort_values(by='similarity', ascending=False).head(5)
+
+def sort_by_nearest_location(df, user_lat, user_lon):
+    df['distance_km'] = df.apply(
+        lambda row: geodesic((user_lat, user_lon), (row['latitude'], row['longitude'])).km,
+        axis=1
+    )
+    return df.sort_values(by='distance_km')
+
+# --- Fungsi Utama Gradio ---
+def wisata_rekomendasi(deskripsi, lokasi):
+    if df.empty:
+        return "Data wisata tidak tersedia. Pastikan cleaned_data.csv ada."
+
+    keywords = extract_keywords(deskripsi)
+    if "Error:" in str(keywords):
+        return str(keywords[0])
+    
+    user_description_joined = " ".join(keywords)
+    lat, lon = get_coordinates_from_location(lokasi)
+    if lat is None or lon is None:
+        return "Lokasi tidak ditemukan."
+
+    reverse_location = get_location_name_from_coordinates(lat, lon)
+    top_place = prepare_and_recommend(df.copy(), user_description_joined)
+    sorted_place = sort_by_nearest_location(top_place, lat, lon)
+
+    hasil = f"Lokasi terdekat dari {reverse_location}:\n\n"
+    hasil += sorted_place[['nama', 'distance_km', 'description']].to_string(index=False)
+    return hasil
+
+# --- UI Gradio ---
+demo = gr.Interface(
+    fn=wisata_rekomendasi,
+    inputs=[
+        gr.Textbox(label="Deskripsi Wisata yang Anda Inginkan"),
+        gr.Textbox(label="Lokasi Anda (Contoh: Gunung Kidul, Yogyakarta)")
+    ],
+    outputs="text",
+    title="Sistem Rekomendasi Wisata",
+    description="Masukkan deskripsi dan lokasi, lalu dapatkan rekomendasi tempat wisata terdekat"
+)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()

requirements.txt

@@ -0,0 +1,6 @@
+gradio
+pandas
+scikit-learn
+geopy
+google-generativeai
+python-dotenv