Menyesuaikan app.py dan menambah lokasi di location.py

app.py

@@ -3,6 +3,13 @@ import numpy as np
 from keras.preprocessing import image
 from Model_Load import load_model_from_files
 from description import description
+from location import location
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from io import BytesIO
+from PIL import Image
+import uvicorn
+from fastapi.middleware.cors import CORSMiddleware
 
 
 # Load model dan label
@@ -25,30 +32,72 @@ def classify_image(img):
     confidence = np.max(pred)
     predicted_label = labels[np.argmax(pred)]
 
-    if confidence < 0.5:
-        return "Tidak dapat dikenali", "Tolong arahkan ke objek yang jelas agar bisa diidentifikasikan. Pastikan anda berada di salah satu tempat Benteng_Vredeburg Candi_Borobudur, Candi_Prambanan, Gedung_Agung_Istana_Kepresidenan_Yogyakarta, Masjid_Gedhe_Kauman, Monumen_Serangan_1Maret, Museum_Gunungapi_Merapi, Situs_Ratu_Boko, Taman_Sari, Tugu_Yogyakarta"
+    akurasi = float(confidence)
+    if confidence < 0.8:
+        label_output = "Tidak dapat dikenali (Confidence: {:.2f}%)".format(confidence * 100)
+        deskripsi = (
+            "Tolong arahkan ke objek yang jelas agar bisa diidentifikasikan. "
+            "Pastikan anda berada di salah satu tempat seperti:\n"
+            "- Benteng Vredeburg\n- Candi Borobudur\n- Candi Prambanan\n"
+            "- Gedung Agung Istana Kepresidenan Yogyakarta\n- Masjid Gedhe Kauman\n"
+            "- Monumen Serangan 1 Maret\n- Museum Gunungapi Merapi\n- Situs Ratu Boko\n"
+            "- Taman Sari\n- Tugu Yogyakarta"
+        )
+        lokasi = "-"
     else:
-        deskripsi = description.get(predicted_label, "Deskripsi belum tersedia.")
         label_output = f"{predicted_label} (Confidence: {confidence * 100:.2f}%)"
-        return label_output, deskripsi
+        deskripsi = description.get(predicted_label, "Deskripsi belum tersedia.")
+        lokasi = location.get(predicted_label, None)
+        if lokasi:
+            lokasi = f'<a href="{lokasi}" target="_blank">Lihat Lokasi di Google Maps</a>'
+        else:
+            lokasi = "Lokasi tidak ditemukan"
 
+    return label_output, deskripsi, lokasi, akurasi
 
 
+# FastAPI instance
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["http://localhost:9000"],  # atau sesuaikan dengan asal frontend
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-# Buat antarmuka Gradio
-demo = gr.Interface(
+@app.post("/api/predict")
+async def predict(file: UploadFile = File(...)):
+    contents = await file.read()
+    img = Image.open(BytesIO(contents)).convert("RGB")
+    label_output, deskripsi, lokasi, akurasi = classify_image(img)
+    return JSONResponse(content={
+        "label_output": label_output,
+        "deskripsi": deskripsi,
+        "lokasi" : lokasi,
+        "confidence": akurasi
+    })
+
+
+# Gradio antarmuka (opsional tetap ditampilkan)
+gradio_app = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil", label="Upload Gambar"),
     outputs=[
         gr.Textbox(label="Output Klasifikasi"),
-        gr.Textbox(label="Deskripsi Lengkap", lines=20, max_lines=50)
+        gr.Textbox(label="Deskripsi Lengkap", lines=20, max_lines=50),
+        gr.HTML(label="Link Lokasi"),
     ],
     allow_flagging="never",
     title="Klasifikasi Gambar",
-    description="Upload gambar kami akan mengklasifikasikan dan memberikan deskripsi mengenai gambar tersebut."
+    description="Upload gambar, sistem akan mengklasifikasikan dan memberikan deskripsi mengenai gambar tersebut."
 )
 
+# Mount Gradio ke FastAPI
+app = gr.mount_gradio_app(app, gradio_app, path="")
 
-# Launch untuk Hugging Face Spaces
+# Jalankan app
 if __name__ == "__main__":
-    demo.launch()
+    uvicorn.run(app, host="127.0.0.1", port=8000)
+

location.py

@@ -0,0 +1,12 @@
+location = {
+    "Benteng Vredeburg": "https://surli.cc/bffath",
+    "Candi Borobudur": "https://surl.lu/rqoiya",
+    "Candi Prambanan": "https://surl.lu/evjzzc",
+    "Gedung Agung Istana Kepresidenan": "https://surl.li/czgmti",
+    "Masjid Gedhe Kauman": "https://surl.li/wpatgm",
+    "Monumen Serangan 1 Maret": "https://surl.lu/axmfwy",
+    "Museum Gunungapi Merapi": "https://surl.lu/wnkqof",
+    "Situs Ratu Boko": "https://surl.li/abdxrb",
+    "Taman Sari": "https://surl.li/rwpdyu",
+    "Tugu Yogyakarta": "https://surl.li/tnehjk"
+}