Update app.py

app.py

@@ -3,8 +3,7 @@ import uuid
 import requests
 import threading
 import time
-from flask import Flask, request, jsonify, render_template
-# استفاده از کتابخانه مستقیم ترجمه به جای کلاینت گرادیو
+from flask import Flask, request, jsonify, render_template, send_from_directory
 from deep_translator import GoogleTranslator
 
 app = Flask(__name__)
@@ -16,11 +15,10 @@ SPACE_HOST = os.environ.get("SPACE_HOST", "")
 SPACE_URL = f"https://{SPACE_HOST}"
 
 def delete_github_run(gh_run_id):
-    time.sleep(15)
+    time.sleep(20)
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/actions/runs/{gh_run_id}"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
     requests.delete(url, headers=headers)
-    print(f"Action {gh_run_id} deleted!")
 
 @app.route('/')
 def index():
@@ -28,84 +26,91 @@ def index():
 
 @app.route('/<page_name>')
 def serve_html(page_name):
+    if not page_name.endswith('.html'):
+        page_name += '.html'
     try:
-        if not page_name.endswith('.html'):
-            page_name += '.html'
         return render_template(page_name)
     except:
         return "Page not found", 404
 
+# --- API برای ساخت عکس (Flux و Cartoon) ---
 @app.route('/api/generate', methods=['POST'])
 def generate():
-    if not GITHUB_TOKEN:
-        return jsonify({"status": "error", "message": "Secret GITHUB_TOKEN is missing!"}), 500
-
+    if not GITHUB_TOKEN: return jsonify({"error": "Token missing"}), 500
     data = request.json
     persian_prompt = data.get('prompt', '')
-    width = data.get('width', 1024)
-    height = data.get('height', 1024)
-    action_name = data.get('action_name', 'generate-flux')
-    
-    english_prompt = persian_prompt
-    
-    # 1. ترجمه متن بسیار سریع داخل خود سرور داکر
-    try:
-        print(f"Translating: {persian_prompt}")
-        translated = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
-        if translated:
-            english_prompt = translated
-        print(f"Translated to: {english_prompt}")
-    except Exception as e:
-        print("Translation error:", e)
-        # در صورت بروز هرگونه خطا، از همان متن اصلی استفاده می‌شود
+    try: english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
+    except: english_prompt = persian_prompt
 
-    # 2. ارسال دستور به گیت‌هاب اکشن
     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}"
+    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}
     }
+    r = requests.post(url, headers=headers, json=payload)
+    return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+
+# --- API جدید مخصوص ویرایش عکس (AI Photoshop) ---
+@app.route('/api/edit', methods=['POST'])
+def edit_image():
+    if not GITHUB_TOKEN: return jsonify({"error": "Token missing"}), 500
     
+    # گرفتن فایل و متن از فرم
+    if 'image' not in request.files: return jsonify({"error": "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": action_name,
+        "event_type": "generate-editor",
         "client_payload": {
             "prompt": english_prompt,
-            "width": width,
-            "height": height,
+            "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_prompt": english_prompt})
-    else:
-        return jsonify({"status": "error", "message": r.text}), 400
+    requests.post(url, headers=headers, json=payload)
+    return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
 
+# وب‌هوک برای دریافت فایل نهایی از گیت‌هاب
 @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")
-    
-    if gh_run_id:
-        threading.Thread(target=delete_github_run, args=(gh_run_id,)).start()
-        
+    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
 
+# چک کردن وضعیت توسط مرورگر
 @app.route('/api/status/<run_id>')
 def check_status(run_id):
-    file_path = f"static/images/{run_id}.webp"
-    if os.path.exists(file_path):
-        return jsonify({"status": "ready", "url": f"/{file_path}"})
+    path = f"static/images/{run_id}.webp"
+    if os.path.exists(path):
+        return jsonify({"status": "ready", "url": f"/{path}"})
     return jsonify({"status": "processing"})
 
+# دسترسی دادن به پوشه تصاویر
+@app.route('/static/images/<filename>')
+def serve_image(filename):
+    return send_from_directory('static/images', filename)
+
 if __name__ == '__main__':
     os.makedirs("static/images", exist_ok=True)
     app.run(host='0.0.0.0', port=7860)