Update app.py

app.py

@@ -1,78 +1,74 @@
-# --- 1. ייבוא הספריות הנדרשות ---
-from flask import Flask, request, send_file, jsonify
-from flask_cors import CORS
-from rembg import remove
-from PIL import Image
-import io
-import os
-
-# --- 2. יצירת אפליקציית Flask ---
-# זוהי היצירה של אובייקט השרת המרכזי שלנו.
-app = Flask(__name__)
-
-# --- 3. הגדרת CORS ---
-# זהו החלק הקריטי המאפשר לאתר ה-Front-End שלנו (שיושב בדומיין אחר)
-# לתקשר עם השרת הזה. בלעדיו, הדפדפן יחסום את הבקשות מטעמי אבטחה.
-# הגדרת "origins='*'" היא פחות מאובטחת ומתאימה לפיתוח.
-# בסביבת ייצור אמיתית, כדאי להחליף את '*' בכתובת המדויקת של אתר ה-Front-End.
-CORS(app, resources={r"/api/*": {"origins": "*"}})
-
-# --- 4. הגדרת "נקודת קצה" (Endpoint) לבדיקת תקינות ---
-# זוהי נקודת קצה פשוטה שנוכל לגשת אליה מהדפדפן כדי לוודא שהשרת פועל.
-# לדוגמה: https://your-backend-url.com/
-@app.route('/')
-def index():
-    # מחזיר הודעת JSON פשוטה
-    return jsonify({"message": "Background removal server is running!"})
-
-# --- 5. הגדרת נקודת הקצה המרכזית להסרת רקע ---
-# כתובת זו תקבל את בקשות ה-POST עם קבצי התמונה.
-@app.route('/api/remove-background', methods=['POST'])
-def remove_background_api():
-    # --- 5.1. בדיקה שהקובץ אכן נשלח בבקשה ---
-    if 'file' not in request.files:
-        # אם המפתח 'file' לא קיים, החזר שגיאה ברורה.
-        return jsonify({"error": "No file part in the request"}), 400
-
-    file = request.files['file']
-
-    # --- 5.2. בדיקה שהמשתמש אכן בחר קובץ ---
-    if file.filename == '':
-        # אם נשלח מפתח 'file' אבל הוא ריק, החזר שגיאה.
-        return jsonify({"error": "No file selected for uploading"}), 400
-
-    # --- 5.3. קריאת הבייטים של הקובץ מהבקשה ---
-    # `file.read()` קורא את כל תוכן הקובץ לתוך משתנה בזיכרון.
-    input_bytes = file.read()
-
-    try:
-        # --- 5.4. שימוש ב-rembg להסרת הרקע ---
-        # זוהי שורת הקסם. הפונקציה `remove` מקבלת את הבייטים של התמונה המקורית
-        # ומחזירה את הבייטים של תמונת ה-PNG החדשה, ללא רקע.
-        output_bytes = remove(input_bytes)
-
-        # --- 5.5. שליחת התמונה המעובדת בחזרה למשתמש ---
-        # אנו אורזים את הבייטים של התוצאה באובייקט BytesIO, שמתנהג כמו קובץ בזיכרון.
-        # `send_file` היא פונקציית עזר של Flask שיודעת לשלוח קבצים בצורה נכונה,
-        # עם ההגדרות (Headers) המתאימות כדי שהדפדפן יבין שזהו קובץ תמונה.
-        return send_file(
-            io.BytesIO(output_bytes),
-            mimetype='image/png', # חשוב להגדיר את סוג התוכן כתמונת PNG
-            as_attachment=False, # כדי שהדפדפן יציג אותה ולא ינסה להוריד ישר
-            download_name='processed.png' # שם ברירת מחדל אם המשתמש ירצה לשמור
-        )
-
-    except Exception as e:
-        # --- 5.6. טיפול בשגיאות אפשריות במהלך העיבוד ---
-        # למשל, אם rembg לא הצליח לעבד קובץ פגום.
-        print(f"Error processing image: {e}") # הדפסת השגיאה ללוג של השרת
-        return jsonify({"error": f"Failed to process image: {str(e)}"}), 500
-
-# --- 6. הרצת השרת (רלוונטי רק לפיתוח מקומי) ---
-# החלק הזה מאפשר לנו להריץ את השרת על המחשב שלנו לצורך בדיקות.
-# פלטפורמות האחסון (כמו Hugging Face) לא משתמשות בחלק הזה,
-# אלא מריצות את האפליקציה באמצעות שרת ייעודי כמו Gunicorn.
-if __name__ == '__main__':
-    # מאזין בכל הכתובות (0.0.0.0) על פורט שמוגדר על ידי סביבת העבודה, או 8080.
-    port = int(os.environ.get('PORT', 8080))
-    app.run(host='0.0.0.0', port=port)
+# --- 1. ייבוא הספריות הנדרשות ---
+from flask import Flask, request, send_file, jsonify
+from flask_cors import CORS
+from rembg import remove
+from PIL import Image
+import io
+import os
+
+# --- 2. יצירת אפליקציית Flask ---
+# זוהי היצירה של אובייקט השרת המרכזי שלנו.
+app = Flask(__name__)
+
+# --- 3. הגדרת CORS ---
+# זהו החלק הקריטי המאפשר לאתר ה-Front-End שלנו (שיושב בדומיין אחר)
+# לתקשר עם השרת הזה. בלעדיו, הדפדפן יחסום את הבקשות מטעמי אבטחה.
+# הגדרת "origins='*'" היא פחות מאובטחת ומתאימה לפיתוח.
+# בסביבת ייצור אמיתית, כדאי להחליף את '*' בכתובת המדויקת של אתר ה-Front-End.
+CORS(app, resources={r"/api/*": {"origins": "*"}})
+
+# --- 4. הגדרת "נקודת קצה" (Endpoint) לבדיקת תקינות ---
+# זוהי נקודת קצה פשוטה שנוכל לגשת אליה מהדפדפן כדי לוודא שהשרת פועל.
+# לדוגמה: https://your-backend-url.com/
+@app.route('/')
+def index():
+    # מחזיר הודעת JSON פשוטה
+    return jsonify({"message": "Background removal server is running!"})
+
+# --- 5. הגדרת נקודת הקצה המרכזית להסרת רקע ---
+# כתובת זו תקבל את בקשות ה-POST עם קבצי התמונה.
+@app.route('/api/remove-background', methods=['POST'])
+def remove_background_api():
+    # --- 5.1. בדיקה שהקובץ אכן נשלח בבקשה ---
+    if 'file' not in request.files:
+        # אם המפתח 'file' לא קיים, החזר שגיאה ברורה.
+        return jsonify({"error": "No file part in the request"}), 400
+
+    file = request.files['file']
+
+    # --- 5.2. בדיקה שהמשתמש אכן בחר קובץ ---
+    if file.filename == '':
+        # אם נשלח מפתח 'file' אבל הוא ריק, החזר שגיאה.
+        return jsonify({"error": "No file selected for uploading"}), 400
+
+    # --- 5.3. קריאת הבייטים של הקובץ מהבקשה ---
+    # `file.read()` קורא את כל תוכן הקובץ לתוך משתנה בזיכרון.
+    input_bytes = file.read()
+
+    try:
+        # --- 5.4. שימוש ב-rembg להסרת הרקע ---
+        # זוהי שורת הקסם. הפונקציה `remove` מקבלת את הבייטים של התמונה המקורית
+        # ומחזירה את הבייטים של תמונת ה-PNG החדשה, ללא רקע.
+        output_bytes = remove(input_bytes)
+
+        # --- 5.5. שליחת התמונה המעובדת בחזרה למשתמש ---
+        # אנו אורזים את הבייטים של התוצאה באובייקט BytesIO, שמתנהג כמו קובץ בזיכרון.
+        # `send_file` היא פונקציית עזר של Flask שיודעת לשלוח קבצים בצורה נכונה,
+        # עם ההגדרות (Headers) המתאימות כדי שהדפדפן יבין שזהו קובץ תמונה.
+        return send_file(
+            io.BytesIO(output_bytes),
+            mimetype='image/png', # חשוב להגדיר את סוג התוכן כתמונת PNG
+            as_attachment=False, # כדי שהדפדפן יציג אותה ולא ינסה להוריד ישר
+            download_name='processed.png' # שם ברירת מחדל אם המשתמש ירצה לשמור
+        )
+
+    except Exception as e:
+        # --- 5.6. טיפול בשגיאות אפשריות במהלך העיבוד ---
+        # למשל, אם rembg לא הצליח לעבד קובץ פגום.
+        print(f"Error processing image: {e}") # הדפסת השגיאה ללוג של השרת
+        return jsonify({"error": f"Failed to process image: {str(e)}"}), 500
+
+# --- 6. הרצת השרת (רלוונטי רק לפיתוח מקומי) ---
+# החלק הזה מאפשר לנו להריץ את השרת על המחשב שלנו לצורך בדיקות.
+# פלטפורמות האחסון (כמו Hugging Face) לא משתמשות בחלק הזה,
+# אלא מריצות את האפליקציה באמצעות שרת ייעודי כמו Gunicorn.