Update app.py

app.py

@@ -2,14 +2,12 @@ import gradio as gr
 from ultralytics import YOLO
 from collections import Counter
 import os
-from glob import glob
-from PIL import Image
 import uuid
 
-# Modeli yükle
+# Model yükle
 model = YOLO("best.pt")
 
-# Sınıf isimleri
+# Sınıf eşlemesi
 id2name = {
     0: "Silah", 1: "Tank", 2: "Helikopter", 3: "Savaş Uçağı", 4: "Asker",
     5: "Top", 6: "Füze", 7: "Savaş Gemisi", 8: "Askeri Araç", 9: "Bayrak",
@@ -18,65 +16,96 @@ id2name = {
 
 PRED_DIR = "yolo_outputs"
 
-def predict(image):
+def save_and_predict(image):
     os.makedirs("inputs", exist_ok=True)
     image_id = str(uuid.uuid4())[:8]
     image_path = f"inputs/{image_id}.jpg"
     image.save(image_path)
 
-    # YOLO tahmini + save_txt
     model.predict(source=image_path, save_txt=True, conf=0.25, project=PRED_DIR, name=image_id, exist_ok=True)
 
-    # TXT yolunu tanımla
     txt_path = f"{PRED_DIR}/{image_id}/labels/{image_id}.txt"
 
     if not os.path.exists(txt_path):
-        return "Hiçbir nesne tespit edilmedi.", "", "", "", "", "⚠️ .txt dosyası oluşmadı"
+        return {}, "Hiçbir nesne tespit edilemedi."
 
-    # TXT dosyasını oku ve class_id çıkar
-    class_ids = []
     with open(txt_path, "r") as f:
-        lines = f.readlines()
-        debug_txt = "".join(lines)
-        for line in lines:
-            class_id = int(line.strip().split()[0])
-            class_ids.append(class_id)
+        class_ids = [int(line.strip().split()[0]) for line in f]
 
     counts = Counter(class_ids)
     detected_names = {id2name[i] for i in counts if i in id2name}
 
-    # Sorular
-    cevap1_lines = [f"{counts[i]} adet {id2name[i]}" for i in counts]
-    cevap1 = f"{len(counts)} farklı sınıf tespit edildi:\n" + "\n".join(cevap1_lines)
+    data = {
+        "counts": counts,
+        "names": detected_names
+    }
 
-    savunma = {"Tank", "Füze", "Top", "Helikopter"}
-    cevap2 = f"Bulunan savunma sistemleri: {', '.join(savunma & detected_names)}" if savunma & detected_names else "Savunma sistemi tespit edilmedi."
+    return data, "Tahmin başarıyla yüklendi. Sağdaki kutulardan bir soruya tıklayın."
 
-    cevap3 = "Askeri birlik olabilir." if {"Asker", "Askeri Araç"} & detected_names else "Askeri birlik tespit edilmedi."
+def cevapla(soru_id, data, chat_history):
+    if not data:
+        return chat_history + [("❓ " + soru_id, "Önce bir görsel yükleyin ve tahmin yapın.")]
 
-    hava = {"Helikopter", "Savaş Uçağı"}
-    cevap4 = f"Hava unsurları tespit edildi: {', '.join(hava & detected_names)}" if hava & detected_names else "Hava unsuru tespit edilmedi."
+    counts = data["counts"]
+    names = data["names"]
+
+    if soru_id == "1️⃣ Kaç nesne ve kaçar adet?":
+        detaylar = [f"{counts[i]} adet {id2name[i]}" for i in counts]
+        cevap = f"{len(counts)} farklı sınıf tespit edildi:\n" + "\n".join(detaylar)
+
+    elif soru_id == "2️⃣ Savunma sistemleri var mı?":
+        savunma = {"Tank", "Füze", "Top", "Helikopter"}
+        bulunan = savunma & names
+        cevap = f"Bulunan savunma sistemleri: {', '.join(bulunan)}" if bulunan else "Savunma sistemi tespit edilmedi."
+
+    elif soru_id == "3️⃣ Askeri birlik tespiti?":
+        cevap = "Askeri birlik olabilir." if {"Asker", "Askeri Araç"} & names else "Askeri birlik tespit edilmedi."
+
+    elif soru_id == "4️⃣ Hava unsurları var mı?":
+        hava = {"Helikopter", "Savaş Uçağı"}
+        bulunan = hava & names
+        cevap = f"Hava unsurları tespit edildi: {', '.join(bulunan)}" if bulunan else "Hava unsuru tespit edilmedi."
+
+    elif soru_id == "5️⃣ Tehdit seviyesi nedir?":
+        yüksek = {"Tank", "Füze", "Savaş Uçağı", "Top", "Savaş Gemisi"}
+        if yüksek & names:
+            cevap = "Yüksek tehdit içeriyor."
+        elif "Asker" in names:
+            cevap = "Orta seviye tehdit."
+        else:
+            cevap = "Düşük tehdit seviyesi."
 
-    yüksek_tehdit = {"Tank", "Füze", "Savaş Uçağı", "Top", "Savaş Gemisi"}
-    if yüksek_tehdit & detected_names:
-        cevap5 = "Yüksek tehdit içeriyor."
-    elif "Asker" in detected_names:
-        cevap5 = "Orta seviye tehdit."
     else:
-        cevap5 = "Düşük tehdit seviyesi."
-
-    return cevap1, cevap2, cevap3, cevap4, cevap5, debug_txt
-gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="pil", label="Görsel Yükle"),
-    outputs=[
-        gr.Textbox(label="1️⃣ Kaç nesne ve kaçar adet?"),
-        gr.Textbox(label="2️⃣ Savunma sistemleri var mı?"),
-        gr.Textbox(label="3️⃣ Askeri birlik tespiti?"),
-        gr.Textbox(label="4️⃣ Hava unsurları var mı?"),
-        gr.Textbox(label="5️⃣ Tehdit seviyesi nedir?"),
-        gr.Textbox(label="🧾 .txt içeriği (debug amaçlı)")
-    ],
-    title="🪖 YOLO Tahmin + Sınıf Analizi + .txt Görüntüleme",
-    description="YOLO tahmini sonrası oluşturulan .txt dosyasını gösterir ve analiz eder."
-).launch()
+        cevap = "Geçersiz soru."
+
+    return chat_history + [(soru_id, cevap)]
+
+# Gradio Blocks yapısı
+with gr.Blocks() as demo:
+    gr.Markdown("## 🪖 YOLO Görselden Nesne Tespiti + Soru-Cevap Arayüzü")
+
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Görsel Yükle")
+            tahmin_bilgisi = gr.Textbox(label="Tahmin Durumu")
+            hidden_data = gr.State()
+            upload_button = gr.Button("📤 Tahmin Yap")
+        with gr.Column():
+            chatbot = gr.Chatbot(label="📡 Soru-Cevap Paneli")
+            with gr.Row():
+                b1 = gr.Button("1️⃣ Kaç nesne ve kaçar adet?")
+                b2 = gr.Button("2️⃣ Savunma sistemleri var mı?")
+            with gr.Row():
+                b3 = gr.Button("3️⃣ Askeri birlik tespiti?")
+                b4 = gr.Button("4️⃣ Hava unsurları var mı?")
+            b5 = gr.Button("5️⃣ Tehdit seviyesi nedir?")
+
+    # Buton olayları
+    upload_button.click(save_and_predict, inputs=image_input, outputs=[hidden_data, tahmin_bilgisi])
+    b1.click(cevapla, inputs=["b1", hidden_data, chatbot], outputs=chatbot).then(fn=lambda: "1️⃣ Kaç nesne ve kaçar adet?", outputs="b1")
+    b2.click(cevapla, inputs=["b2", hidden_data, chatbot], outputs=chatbot).then(fn=lambda: "2️⃣ Savunma sistemleri var mı?", outputs="b2")
+    b3.click(cevapla, inputs=["b3", hidden_data, chatbot], outputs=chatbot).then(fn=lambda: "3️⃣ Askeri birlik tespiti?", outputs="b3")
+    b4.click(cevapla, inputs=["b4", hidden_data, chatbot], outputs=chatbot).then(fn=lambda: "4️⃣ Hava unsurları var mı?", outputs="b4")
+    b5.click(cevapla, inputs=["b5", hidden_data, chatbot], outputs=chatbot).then(fn=lambda: "5️⃣ Tehdit seviyesi nedir?", outputs="b5")
+
+demo.launch()