Create app.py

app.py

@@ -0,0 +1,80 @@
+import os
+import uuid
+import requests
+from flask import Flask, request, jsonify, render_template
+
+app = Flask(__name__)
+
+# خواندن اطلاعات از Secrets های اسپیس
+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}"
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/api/generate', methods=['POST'])
+def generate():
+    if not GITHUB_TOKEN:
+        return jsonify({"status": "error", "message": "توکن گیت‌هاب در Secrets یافت نشد!"}), 500
+
+    data = request.json
+    prompt = data.get('prompt')
+    width = data.get('width', '1024')
+    height = data.get('height', '1024')
+    
+    # تولید یک شناسه یکتا برای این درخواست
+    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": "generate-image",
+        "client_payload": {
+            "prompt": prompt,
+            "width": width,
+            "height": height,
+            "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})
+    else:
+        return jsonify({"status": "error", "message": r.text}), 400
+
+@app.route('/api/webhook/upload', methods=['POST'])
+def webhook_upload():
+    # گیت‌هاب وقتی عکس را ساخت، آن را با فرمت POST به این آدرس می‌فرستد
+    run_id = request.form.get('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")
+    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}"})
+    return jsonify({"status": "processing"})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)