Update app.py

app.py

@@ -36,65 +36,90 @@ def serve_html(page_name):
 # --- API برای ساخت عکس (Flux و Cartoon) ---
 @app.route('/api/generate', methods=['POST'])
 def generate():
-    if not GITHUB_TOKEN: return jsonify({"error": "Token missing"}), 500
+    if not GITHUB_TOKEN: 
+        return jsonify({"status": "error", "message": "Secret GITHUB_TOKEN is missing!"}), 500
+        
     data = request.json
     persian_prompt = data.get('prompt', '')
-    try: english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
-    except: english_prompt = persian_prompt
+    
+    try: 
+        english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
+    except: 
+        english_prompt = persian_prompt
 
     run_id = str(uuid.uuid4())
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
+    
     payload = {
         "event_type": data.get('action_name', 'generate-flux'),
-        "client_payload": {"prompt": english_prompt, "width": data.get('width', 1024), "height": data.get('height', 1024), "run_id": run_id, "space_url": SPACE_URL}
+        "client_payload": {
+            "prompt": english_prompt, 
+            "width": data.get('width', 1024), 
+            "height": data.get('height', 1024), 
+            "run_id": run_id, 
+            "space_url": SPACE_URL
+        }
     }
     r = requests.post(url, headers=headers, json=payload)
-    return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+    
+    if r.status_code == 204:
+        # کلمه کلیدی translated_prompt برگردانده شد تا با fluxpro.html و cartoon.html کاملا سازگار باشد
+        return jsonify({
+            "status": "success", 
+            "run_id": run_id, 
+            "translated_prompt": english_prompt, 
+            "translated": english_prompt
+        })
+    else:
+        return jsonify({"status": "error", "message": r.text}), 400
 
 # --- API جدید مخصوص ویرایش عکس (AI Photoshop) ---
 @app.route('/api/edit', methods=['POST'])
 def edit_image():
-    if not GITHUB_TOKEN: return jsonify({"error": "Token missing"}), 500
+    if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
     
-    # گرفتن فایل و متن از فرم
-    if 'image' not in request.files: return jsonify({"error": "Image required"}), 400
+    if 'image' not in request.files: return jsonify({"status": "error", "message": "Image required"}), 400
     file = request.files['image']
     persian_prompt = request.form.get('prompt', '')
     
-    # ترجمه متن
     try: english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
     except: english_prompt = persian_prompt
 
     run_id = str(uuid.uuid4())
     
-    # ذخیره عکس اولیه در داکر تا گیت‌هاب بتواند آن را دانلود کند
     input_filename = f"{run_id}_input.png"
     file.save(f"static/images/{input_filename}")
     image_public_url = f"{SPACE_URL}/static/images/{input_filename}"
 
-    # ارسال دستور به گیت‌هاب
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
     payload = {
         "event_type": "generate-editor",
         "client_payload": {
             "prompt": english_prompt,
-            "image_url": image_public_url, # لینک عکسی که در داکر ذخیره کردیم
+            "image_url": image_public_url,
             "run_id": run_id,
             "space_url": SPACE_URL
         }
     }
-    requests.post(url, headers=headers, json=payload)
-    return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+    r = requests.post(url, headers=headers, json=payload)
+    
+    if r.status_code == 204:
+        return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+    else:
+        return jsonify({"status": "error", "message": r.text}), 400
 
 # وب‌هوک برای دریافت فایل نهایی از گیت‌هاب
 @app.route('/api/webhook/upload', methods=['POST'])
 def webhook_upload():
     run_id = request.form.get('run_id')
     gh_run_id = request.form.get('github_run_id')
+    if 'file' not in request.files or not run_id:
+        return "Invalid request", 400
+        
     file = request.files['file']
-    file.save(f"static/images/{run_id}.webp") # فایل نهایی ذخیره می‌شود
+    file.save(f"static/images/{run_id}.webp")
     if gh_run_id: threading.Thread(target=delete_github_run, args=(gh_run_id,)).start()
     return "OK", 200