Update app.py

app.py

@@ -323,16 +323,21 @@ def decrypt_result():
     predictions = []
     for i, enc in enumerate(encrypted_output):
         pred = client.deserialize_decrypt_dequantize(enc)
-        
-        # 🔍 Yeni log ve yuvarlama işlemleri
-        raw_value = pred.flatten()[0]
-        label_idx = int(round(raw_value))
-        
+        raw = pred.flatten()
+
         print(f"\n🔍 [Tahmin {i+1}]")
-        print(f"🔢 Decrypted float değer: {raw_value}")
-        print(f"🔁 Yuvarlanmış index: {label_idx}")
+        print(f"📈 Decrypted output (vektör?): {raw}")
         print(f"🎯 Sınıf listesi: {list(encoder.classes_)}")
 
+        if len(raw) > 1:
+            # Çok sınıflı olasılık çıktısı (örn. [0.1, 0.6, 0.2, 0.1])
+            label_idx = int(np.argmax(raw))
+            print(f"🔢 argmax sonucu: {label_idx}")
+        else:
+            # Tek değer → doğrudan sınıf indexi olabilir
+            label_idx = int(round(raw[0]))
+            print(f"🔢 Yuvarlanmış değer: {raw[0]} → {label_idx}")
+
         if label_idx >= len(encoder.classes_):
             label = f"⚠️ Geçersiz sınıf indexi: {label_idx}"
         else:
@@ -344,6 +349,7 @@ def decrypt_result():
 
 
 
+
     
 
 # -------------------- Gradio UI -------------------- #