Create app.py

app.py

@@ -0,0 +1,141 @@
+import os
+import uuid
+import requests
+import threading
+import time
+from flask import Flask, request, jsonify, render_template, send_from_directory
+from deep_translator import GoogleTranslator
+
+app = Flask(__name__)
+
+GITHUB_TOKEN = os.environ.get("GITHUB_TOKEN")
+GITHUB_USER = os.environ.get("GITHUB_USER", "Hamed744")
+GITHUB_REPO = os.environ.get("GITHUB_REPO", "vid-test")
+SPACE_HOST = os.environ.get("SPACE_HOST", "")
+SPACE_URL = f"https://{SPACE_HOST}"
+
+def delete_github_run(gh_run_id):
+    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)
+
+@app.route('/')
+def index():
+    return render_template('fluxpro.html')
+
+@app.route('/<page_name>')
+def serve_html(page_name):
+    if not page_name.endswith('.html'):
+        page_name += '.html'
+    try:
+        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
+        
+    data = request.json
+    persian_prompt = data.get('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
+        }
+    }
+    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, 
+            "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({"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', '')
+    
+    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,
+            "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')
+    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()
+    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}"})
+    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)