תיקון שגיאות

app.py

@@ -26,17 +26,12 @@ def inference(image: Image.Image, gemini_api_key: str):
 
     try:
         encoded_image = process_image(image, gemini_api_key, progress_callback=progress_callback)
-        # המרת התמונה חזרה לפורמט PIL Image
-        decoded_image = Image.open(BytesIO(base64.b64decode(encoded_image)))
-        # המרת התמונה לפורמט base64 data URL
-        buffered = BytesIO()
-        decoded_image.save(buffered, format="JPEG")
-        img_str = base64.b64encode(buffered.getvalue()).decode()
-        img_str = f"data:image/jpeg;base64,{img_str}"
+        # יצירת מחרוזת base64 data URL
+        img_str = f"data:image/jpeg;base64,{encoded_image}"
 
         # החזרת רשימה עם התמונה היחידה בפורמט base64
         return [img_str], gr.update(value="", visible=False)
-    
+
     except Exception as e:
         # טיפול בשגיאה והחזרת הודעה ב-Textbox
         error_message = f"שגיאה: {type(e).__name__}\n"

backend.py

@@ -8,6 +8,7 @@ import cv2
 from PIL import Image, ImageFilter
 from scipy.ndimage import binary_dilation
 from omegaconf import OmegaConf
+from io import BytesIO
 # -----------------------------
 # 1) הגדרת המפתח API של Gemini כפרמטר
 # -----------------------------
@@ -241,7 +242,7 @@ def process_image(
     pil_image: Image.Image,
     gemini_api_key: str,
     progress_callback=None
-) -> Image.Image:
+) -> str:  # שינוי: החזרת מחרוזת base64
     if not gemini_api_key:
         raise ValueError("מפתח API של Gemini אינו מוזן.")
     """
@@ -333,5 +334,9 @@ def process_image(
 
     progress_callback(1.0, "סיימנו! מחזירים את התוצאה הסופית.")
 
-    # החזרת התמונה כאובייקט PIL Image
-    return final_image
+    # המרת התמונה ל-base64
+    buffered = BytesIO()
+    final_image.save(buffered, format="JPEG")
+    encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+    return encoded_image