Upload app.py

app.py

@@ -11,47 +11,40 @@ from tqdm import tqdm
 def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p", motion_scale=30):
     try:
         # تهيئة نموذج Stable Diffusion
-        model_id = "CompVis/stable-diffusion-v1-4"  # نموذج أخف
-        device = "cpu"
+        model_id = "CompVis/stable-diffusion-v1-4"
         
         pipe = StableDiffusionPipeline.from_pretrained(
             model_id,
-            torch_dtype=torch.float32
-        )
-        pipe = pipe.to(device)
+            torch_dtype=torch.float32,
+            low_memory=True
+        ).to("cpu")
         
         # تحسين الأداء
         pipe.enable_attention_slicing()
-        pipe.enable_model_cpu_offload()
-        
-        # إنشاء مجلد مؤقت
-        temp_dir = tempfile.mkdtemp()
-        frames = []
         
         # تحديد الحجم
         if resolution == "480p":
-            size = (640, 480)  # حجم أصغر للأداء
+            size = (480, 320)  # حجم أصغر للأداء
         elif resolution == "720p":
-            size = (1280, 720)
+            size = (640, 480)
         else:
-            size = (1920, 1080)
+            size = (854, 480)
         
-        # توليد 3 صور فقط
-        num_images = 3
+        # توليد صورتين فقط
+        num_images = 2
         print(f"جاري توليد {num_images} صور...")
         
+        frames = []
         for i in tqdm(range(num_images)):
             # تعديل البرومبت لكل صورة
             current_prompt = text_prompt
             if i == 1:
-                current_prompt += " , slight movement"
-            elif i == 2:
-                current_prompt += " , more movement"
+                current_prompt += " , with movement and motion"
             
             # توليد الصورة
             image = pipe(
                 prompt=current_prompt,
-                num_inference_steps=20,  # تقليل خطوات الاستدلال
+                num_inference_steps=15,  # تقليل خطوات الاستدلال
                 guidance_scale=7.0
             ).images[0]
             
@@ -66,19 +59,19 @@ def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p",
         
         # إنشاء الإطارات الوسيطة
         final_frames = []
-        for i in range(len(frames)-1):
-            # إضافة الإطار الحالي
-            final_frames.append(frames[i])
-            
-            # إنشاء إطارات وسيطة
-            num_transitions = 10  # عدد الإطارات الوسيطة
-            for t in range(num_transitions):
-                alpha = t / num_transitions
-                transition_frame = (1 - alpha) * frames[i] + alpha * frames[i+1]
-                final_frames.append(transition_frame.astype(np.uint8))
+        
+        # إضافة الإطار الأول
+        final_frames.append(frames[0])
+        
+        # إنشاء إطارات وسيطة بين الصورتين
+        num_transitions = 8
+        for t in range(num_transitions):
+            alpha = t / num_transitions
+            transition_frame = (1 - alpha) * frames[0] + alpha * frames[1]
+            final_frames.append(transition_frame.astype(np.uint8))
         
         # إضافة الإطار الأخير
-        final_frames.append(frames[-1])
+        final_frames.append(frames[1])
         
         # تكرار الإطارات للوصول للمدة المطلوبة
         target_frames = int(duration * fps)
@@ -87,8 +80,10 @@ def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p",
         final_frames = final_frames[:target_frames]
         
         # إنشاء الفيديو
-        clip = ImageSequenceClip(final_frames, fps=fps)
+        temp_dir = tempfile.mkdtemp()
         output_path = os.path.join(temp_dir, "output.mp4")
+        
+        clip = ImageSequenceClip(final_frames, fps=fps)
         clip.write_videofile(output_path, codec='libx264', fps=fps)
         
         # تنظيف الذاكرة
@@ -101,11 +96,11 @@ def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p",
     except Exception as e:
         error_msg = str(e)
         print(f"حدث خطأ: {error_msg}")
-        return f"حدث خطأ أثناء إنشاء الفيديو: {error_msg}"
+        return None  # إرجاع None بدلاً من رسالة الخطأ
 
 def video_generator(text_prompt, duration=10, resolution="480p", motion_scale=30):
     if not text_prompt:
-        return "الرجاء إدخال وصف للفيديو"
+        return None
     
     print(f"بدء توليد فيديو متحرك بناءً على الوصف: {text_prompt}")
     print(f"المدة: {duration} ثواني")
@@ -120,30 +115,29 @@ def video_generator(text_prompt, duration=10, resolution="480p", motion_scale=30
         )
         return result
     except Exception as e:
-        error_msg = str(e)
-        print(f"حدث خطأ في المولد: {error_msg}")
-        return f"حدث خطأ: {error_msg}"
+        print(f"حدث خطأ في المولد: {str(e)}")
+        return None
 
 # إنشاء واجهة المستخدم
 iface = gr.Interface(
     fn=video_generator,
     inputs=[
         gr.Textbox(label="وصف المشهد", placeholder="اكتب وصفاً للمشهد المتحرك الذي تريد إنشاءه..."),
-        gr.Slider(minimum=3, maximum=15, value=5, step=1, label="مدة الفيديو (بالثواني)"),
+        gr.Slider(minimum=3, maximum=10, value=3, step=1, label="مدة الفيديو (بالثواني)"),
         gr.Radio(["480p", "720p", "1080p"], label="دقة الفيديو", value="480p"),
         gr.Slider(minimum=10, maximum=100, value=30, step=5, label="مقياس الحركة (%)")
     ],
     outputs=gr.Video(label="الفيديو المتحرك المُنشأ"),
-    title="مولد الفيديو المتحرك بالذكاء الاصطناعي (نسخة سريعة)",
+    title="مولد الفيديو المتحرك بالذكاء الاصطناعي (نسخة خفيفة)",
     description="""
     قم بإدخال وصف للمشهد وسيقوم النظام بإنشاء فيديو متحرك باستخدام الذكاء الاصطناعي.
     
     نصائح للأداء الأفضل:
     - استخدم دقة 480p للحصول على أسرع أداء
-    - اختر مدة قصيرة (3-5 ثواني) للتجربة الأولى
+    - اختر مدة قصيرة (3 ثواني) للتجربة الأولى
     - اكتب وصفاً واضحاً وبسيطاً
     
-    ملاحظة: هذه نسخة مُحسنة للأداء على CPU. تم تبسيط عملية توليد الفيديو للحصول على نتائج أسرع.
+    ملاحظة: هذه نسخة خفيفة جداً تستخدم CPU فقط. قد تستغرق العملية بضع دقائق.
     """,
     theme="huggingface",
     cache_examples=False