חזרה לקוד המקורי התקני

app.py

@@ -25,13 +25,10 @@ def inference(image: Image.Image, gemini_api_key: str):
         progress(fraction, desc=description)
 
     try:
+        # כעת נקרא ל-process_image עם אפשרות לעדכן התקדמות
         encoded_image = process_image(image, gemini_api_key, progress_callback=progress_callback)
-        # יצירת מחרוזת base64 data URL
-        img_str = f"data:image/jpeg;base64,{encoded_image}"
-
-        # החזרת רשימה עם התמונה היחידה בפורמט base64
-        return [img_str], gr.update(value="", visible=False)
-
+        decoded_image = Image.open(BytesIO(base64.b64decode(encoded_image)))
+        return decoded_image, gr.update(value="", visible=False)  # החזרת תמונה ו-Textbox מוסתר
     except Exception as e:
         # טיפול בשגיאה והחזרת הודעה ב-Textbox
         error_message = f"שגיאה: {type(e).__name__}\n"
@@ -62,19 +59,20 @@ demo = gr.Interface(
         )
     ],
     outputs=[
-        gr.Gallery(label="תוצאה סופית"),  # שינוי מ-gr.Image ל-gr.Gallery
-        gr.Textbox(label="שגיאות", visible=False)
+        gr.Image(type="pil", label="תוצאה סופית"),
+        gr.Textbox(label="שגיאות", visible=False)  # הוספת רכיב להצגת שגיאות
     ],
     title=title_str,
     description=description_str,
     examples=[
-        [EXAMPLE_IMAGE]
+        [EXAMPLE_IMAGE]  # תמונה בלבד, ללא מפתח API
     ],
     flagging_mode="never",
-    theme=gr.themes.Default()
+    theme=gr.themes.Default()  # עיצוב קליל לממשק
 )
 
 
 if __name__ == "__main__":
     # ניתן להגדיר share=True אם רוצים לשתף מחוץ לרשת המקומית
-    demo.launch()
+    demo.launch()
+

backend.py

@@ -8,7 +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 כפרמטר
 # -----------------------------
@@ -242,7 +242,7 @@ def process_image(
     pil_image: Image.Image,
     gemini_api_key: str,
     progress_callback=None
-) -> str:  # שינוי: החזרת מחרוזת base64
+) -> Image.Image:
     if not gemini_api_key:
         raise ValueError("מפתח API של Gemini אינו מוזן.")
     """
@@ -335,8 +335,5 @@ def process_image(
     progress_callback(1.0, "סיימנו! מחזירים את התוצאה הסופית.")
 
     # המרת התמונה ל-base64
-    buffered = BytesIO()
-    final_image.save(buffered, format="JPEG")
-    encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
-
+    encoded_image = encode_image_to_base64(final_image)
     return encoded_image