Update app.py

app.py

@@ -64,7 +64,6 @@ def generate():
     r = requests.post(url, headers=headers, json=payload)
     
     if r.status_code == 204:
-        # کلمه کلیدی translated_prompt برگردانده شد تا با fluxpro.html و cartoon.html کاملا سازگار باشد
         return jsonify({
             "status": "success", 
             "run_id": run_id, 
@@ -74,12 +73,12 @@ def generate():
     else:
         return jsonify({"status": "error", "message": r.text}), 400
 
-# --- API جدید مخصوص ویرایش عکس (AI Photoshop) ---
+# --- API مخصوص ویرایش عکس (AI Photoshop) ---
 @app.route('/api/edit', methods=['POST'])
 def edit_image():
     if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
-    
     if 'image' not in request.files: return jsonify({"status": "error", "message": "Image required"}), 400
+    
     file = request.files['image']
     persian_prompt = request.form.get('prompt', '')
     
@@ -110,30 +109,80 @@ def edit_image():
     else:
         return jsonify({"status": "error", "message": r.text}), 400
 
-# وب‌هوک برای دریافت فایل نهایی از گیت‌هاب
+# --- API جدید مخصوص تبدیل عکس به ویدیو (Image to Video) ---
+@app.route('/api/generate-video', methods=['POST'])
+def generate_video():
+    if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
+    if 'image' not in request.files: return jsonify({"status": "error", "message": "Image required"}), 400
+    
+    file = request.files['image']
+    persian_prompt = request.form.get('prompt', '')
+    duration = request.form.get('duration', 5.0)
+    
+    try: 
+        if persian_prompt.strip():
+            english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
+        else:
+            english_prompt = ""
+    except: 
+        english_prompt = persian_prompt
+
+    run_id = str(uuid.uuid4())
+    
+    input_filename = f"{run_id}_video_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-video",
+        "client_payload": {
+            "prompt": english_prompt,
+            "duration": duration,
+            "image_url": image_public_url,
+            "run_id": run_id,
+            "space_url": SPACE_URL
+        }
+    }
+    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')
+    ext = request.form.get('ext', 'webp') # پسوند پیش‌فرض webp برای عکس و mp4 برای ویدیو
+    
     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}.{ext}")
+    
     if gh_run_id: threading.Thread(target=delete_github_run, args=(gh_run_id,)).start()
     return "OK", 200
 
 # چک کردن وضعیت توسط مرورگر
 @app.route('/api/status/<run_id>')
 def check_status(run_id):
-    path = f"static/images/{run_id}.webp"
-    if os.path.exists(path):
-        return jsonify({"status": "ready", "url": f"/{path}"})
+    # چک کردن فایل ویدیویی
+    if os.path.exists(f"static/images/{run_id}.mp4"):
+        return jsonify({"status": "ready", "url": f"/static/images/{run_id}.mp4"})
+    # چک کردن فایل تصویری
+    if os.path.exists(f"static/images/{run_id}.webp"):
+        return jsonify({"status": "ready", "url": f"/static/images/{run_id}.webp"})
+        
     return jsonify({"status": "processing"})
 
-# دسترسی دادن به پوشه تصاویر
+# دسترسی دادن به پوشه تصاویر و ویدیوها
 @app.route('/static/images/<filename>')
-def serve_image(filename):
+def serve_file(filename):
     return send_from_directory('static/images', filename)
 
 if __name__ == '__main__':