Update main.py

main.py

@@ -118,7 +118,8 @@ async def generate_t2v_stream(
         global t2v_pipe
         start_time = time.time()
 
-        yield f"data: {json.dumps({'status': 'info', 'progress': 5, 'message': 'تجهيز بيئة الـ GPU A100...'})}\n\n"
+        payload = json.dumps({'status': 'info', 'progress': 5, 'message': 'تجهيز بيئة الـ GPU A100...'})
+        yield f"data: {payload}\n\n"
 
         if t2v_pipe is None:
             load_models()
@@ -141,7 +142,8 @@ async def generate_t2v_stream(
             print(f"[PROGRESS] {data}")
             return callback_kwargs
 
-        yield f"data: {json.dumps({'status': 'generating', 'progress': 10, 'message': 'بدء عمليات الانتشار التوليدي...'})\n\n"
+        payload = json.dumps({'status': 'generating', 'progress': 10, 'message': 'بدء عمليات الانتشار التوليدي...'})
+        yield f"data: {payload}\n\n"
 
         video_frames = t2v_pipe(
             prompt=prompt,
@@ -153,17 +155,20 @@ async def generate_t2v_stream(
             callback_on_step_end=step_callback
         ).frames[0]
 
-        yield f"data: {json.dumps({'status': 'encoding', 'progress': 92, 'message': 'جاري ضغط وترميز الفيديو بأعلى جودة (MP4)...'})}\n\n"
+        payload = json.dumps({'status': 'encoding', 'progress': 92, 'message': 'جاري ضغط وترميز الفيديو بأعلى جودة (MP4)...'})
+        yield f"data: {payload}\n\n"
 
         filename = f"vid_{uuid.uuid4().hex[:10]}.mp4"
         with tempfile.TemporaryDirectory() as tmpdir:
             temp_video_path = os.path.join(tmpdir, filename)
             export_to_video(video_frames, temp_video_path, fps=8)
 
-            yield f"data: {json.dumps({'status': 'uploading', 'progress': 96, 'message': 'جاري الرفع المباشر إلى Cloudflare R2...'})}\n\n"
+            payload = json.dumps({'status': 'uploading', 'progress': 96, 'message': 'جاري الرفع المباشر إلى Cloudflare R2...'})
+            yield f"data: {payload}\n\n"
             r2_url = upload_file_to_r2(temp_video_path, filename)
 
-        yield f"data: {json.dumps({'status': 'completed', 'progress': 100, 'message': 'تم التوليد بنجاح!', 'video_url': r2_url})}\n\n"
+        payload = json.dumps({'status': 'completed', 'progress': 100, 'message': 'تم التوليد بنجاح!', 'video_url': r2_url})
+        yield f"data: {payload}\n\n"
 
     return StreamingResponse(event_stream(), media_type="text/event-stream")
 
@@ -180,7 +185,8 @@ async def generate_i2v_stream(
         global i2v_pipe
         start_time = time.time()
 
-        yield f"data: {json.dumps({'status': 'info', 'progress': 5, 'message': 'قراءة الصورة وتحضير A100...'})}\n\n"
+        payload = json.dumps({'status': 'info', 'progress': 5, 'message': 'قراءة الصورة وتحضير A100...'})
+        yield f"data: {payload}\n\n"
 
         if i2v_pipe is None:
             load_models()
@@ -201,7 +207,8 @@ async def generate_i2v_stream(
             msg = f"تحريك الصورة: الخطوة {current_step} من {num_inference_steps} | الوقت المتبقي: ~{eta_seconds} ثانية"
             return callback_kwargs
 
-        yield f"data: {json.dumps({'status': 'generating', 'progress': 10, 'message': 'بدء تحريك الإطارات من الصورة...'})}\n\n"
+        payload = json.dumps({'status': 'generating', 'progress': 10, 'message': 'بدء تحريك الإطارات من الصورة...'})
+        yield f"data: {payload}\n\n"
 
         video_frames = i2v_pipe(
             prompt=prompt,
@@ -214,16 +221,19 @@ async def generate_i2v_stream(
             callback_on_step_end=step_callback
         ).frames[0]
 
-        yield f"data: {json.dumps({'status': 'encoding', 'progress': 92, 'message': 'جاري ضغط وترميز الفيديو...'})}\n\n"
+        payload = json.dumps({'status': 'encoding', 'progress': 92, 'message': 'جاري ضغط وترميز الفيديو...'})
+        yield f"data: {payload}\n\n"
 
         filename = f"vid_i2v_{uuid.uuid4().hex[:10]}.mp4"
         with tempfile.TemporaryDirectory() as tmpdir:
             temp_video_path = os.path.join(tmpdir, filename)
             export_to_video(video_frames, temp_video_path, fps=8)
 
-            yield f"data: {json.dumps({'status': 'uploading', 'progress': 96, 'message': 'جاري الرفع إلى Cloudflare R2...'})}\n\n"
+            payload = json.dumps({'status': 'uploading', 'progress': 96, 'message': 'جاري الرفع إلى Cloudflare R2...'})
+            yield f"data: {payload}\n\n"
             r2_url = upload_file_to_r2(temp_video_path, filename)
 
-        yield f"data: {json.dumps({'status': 'completed', 'progress': 100, 'message': 'تم التوليد بنجاح!', 'video_url': r2_url})}\n\n"
+        payload = json.dumps({'status': 'completed', 'progress': 100, 'message': 'تم التوليد بنجاح!', 'video_url': r2_url})
+        yield f"data: {payload}\n\n"
 
     return StreamingResponse(event_stream(), media_type="text/event-stream")