Update comfy_engine.py

comfy_engine.py

@@ -2,6 +2,7 @@ import json, uuid, os, asyncio, time
 import httpx
 import websockets
 from urllib.parse import urlencode
+from datetime import datetime
 
 # AMD GPU सर्वर का IP
 COMFY_HOST = os.getenv("COMFY_HOST", "134.199.132.159")
@@ -17,6 +18,25 @@ RESOLUTIONS = {
     "720p": {"16:9": (1280, 720), "9:16": (720, 1280), "1:1": (768, 768)}
 }
 
+def save_to_history(prompt, video_url):
+    history_file = "history.json"
+    new_entry = {
+        "id": str(uuid.uuid4())[:8],
+        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "prompt": prompt,
+        "url": video_url
+    }
+    data = []
+    if os.path.exists(history_file):
+        try:
+            with open(history_file, "r", encoding="utf-8") as f:
+                data = json.load(f)
+        except: pass
+    
+    data.insert(0, new_entry) # नया वीडियो सबसे ऊपर
+    with open(history_file, "w", encoding="utf-8") as f:
+        json.dump(data[:50], f, indent=4) # सिर्फ लेटेस्ट 50 वीडियोस सेव करेंगे
+
 def inject_params(req: dict) -> dict:
     p = json.loads(json.dumps(WORKFLOW_TEMPLATE))
     
@@ -39,7 +59,6 @@ def inject_params(req: dict) -> dict:
     p["78"]["inputs"]["steps"] = 4 
     p["78"]["inputs"]["cfg"] = 5.0
     
-    # नोट: सीड (Seed) को हमने फिक्स रखा है जैसा कि आपने चाहा था (Consistency के लिए)
     return p
 
 def extract_video_url(history: dict, token: str) -> str:
@@ -72,14 +91,14 @@ async def generate_video_stream(req: dict):
     start_t = time.time()
     progress_fake = 0
 
-    # WebSocket के ज़रिए प्रोग्रेस ट्रैक करना (Bytes Hack शामिल है)
+    # WebSocket के ज़रिए प्रोग्रेस ट्रैक करना
     ws_url = f"ws://{COMFY_HOST}/ws?clientId={client_id}&token={token}"
     try:
         async with websockets.connect(ws_url, ping_interval=20) as ws:
             while True:
                 msg_raw = await ws.recv()
                 
-                # बाइनरी डेटा (Bytes) हैंडलिंग - आपका रेडियो ऐप वाला हैक
+                # बाइनरी डेटा (Bytes) हैंडलिंग - हैक
                 if isinstance(msg_raw, (bytes, bytearray)):
                     if progress_fake < 95 and (time.time() - start_t) > 2:
                         progress_fake = min(95, progress_fake + 1)
@@ -106,7 +125,10 @@ async def generate_video_stream(req: dict):
     
     try:
         v_url = extract_video_url(history, token)
+        
+        # 🚀 MAGIC FIX: यहाँ वीडियो हिस्ट्री में सेव हो रहा है!
+        save_to_history(req.get("prompt", ""), v_url)
+        
         yield f"data: {json.dumps({'progress': 100, 'status': 'done', 'url': v_url})}\n\n"
     except:
         yield f"data: {json.dumps({'error': 'File extraction failed'})}\n\n"
-