Upload 3 files

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline
 from PIL import Image
 import numpy as np
 import cv2
@@ -9,48 +9,55 @@ import os
 import tempfile
 
 def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p"):
-    # تهيئة نموذج Stable Diffusion
-    model_id = "runwayml/stable-diffusion-v1-5"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    if torch.cuda.is_available():
-        pipe = pipe.to("cuda")
-    
-    # إنشاء مجلد مؤقت لحفظ الصور
-    temp_dir = tempfile.mkdtemp()
-    frames = []
-    
-    # تحديد حجم الصورة حسب الدقة المطلوبة
-    if resolution == "480p":
-        size = (854, 480)
-    elif resolution == "720p":
-        size = (1280, 720)
-    else:  # 1080p
-        size = (1920, 1080)
-    
     try:
+        # تهيئة نموذج لتوليد الصور
+        pipe = DiffusionPipeline.from_pretrained(
+            "runwayml/stable-diffusion-v1-5",
+            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+        )
+        
+        if torch.cuda.is_available():
+            pipe = pipe.to("cuda")
+        
+        # إنشاء مجلد مؤقت لحفظ الصور
+        temp_dir = tempfile.mkdtemp()
+        frames = []
+        
+        # تحديد حجم الصورة حسب الدقة المطلوبة
+        if resolution == "480p":
+            size = (854, 480)
+        elif resolution == "720p":
+            size = (1280, 720)
+        else:  # 1080p
+            size = (1920, 1080)
+        
         # توليد عدة صور للحصول على تأثير الحركة
-        num_frames = int(duration * fps / 10)  # نقلل عدد الإطارات للأداء
+        num_frames = max(int(duration * fps / 10), 1)  # على الأقل إطار واحد
+        
         for i in range(num_frames):
             # توليد صورة
-            image = pipe(text_prompt).images[0]
+            with torch.autocast("cuda" if torch.cuda.is_available() else "cpu"):
+                image = pipe(text_prompt, num_inference_steps=20).images[0]
             
             # تحويل حجم الصورة
-            image = image.resize(size)
+            image = image.resize(size, Image.LANCZOS)
             
             # تكرار الإطار عدة مرات للحصول على فيديو أنعم
+            frame = np.array(image)
             for _ in range(10):  # كل صورة تتكرر 10 مرات
-                frames.append(np.array(image))
+                frames.append(frame)
         
         # إنشاء الفيديو من الإطارات
         clip = ImageSequenceClip(frames, fps=fps)
         
         # حفظ الفيديو في ملف مؤقت
         output_path = os.path.join(temp_dir, "output.mp4")
-        clip.write_videofile(output_path, codec='libx264', fps=fps)
+        clip.write_videofile(output_path, codec='libx264', fps=fps, verbose=False, logger=None)
         
         return output_path
         
     except Exception as e:
+        print(f"Error: {str(e)}")
         return str(e)
     finally:
         # تحرير الذاكرة
@@ -58,13 +65,15 @@ def create_video_from_text(text_prompt, duration=10, fps=30, resolution="480p"):
             torch.cuda.empty_cache()
 
 def video_generator(text_prompt, duration=10, resolution="480p"):
+    if not text_prompt:
+        return "الرجاء إدخال وصف للفيديو"
+    
     try:
-        video_path = create_video_from_text(
+        return create_video_from_text(
             text_prompt,
             duration=duration,
             resolution=resolution
         )
-        return video_path
     except Exception as e:
         return str(e)
 
@@ -81,9 +90,9 @@ iface = gr.Interface(
     description="قم بإدخال وصف للفيديو الذي تريد إنشاءه وسيقوم النظام بتوليده باستخدام الذكاء الاصطناعي",
     theme="huggingface",
     examples=[
-        ["شروق الشمس على شاطئ البحر", 10, "720p"],
-        ["مدينة مستقبلية في الليل", 15, "1080p"],
-        ["حديقة جميلة مليئة بالزهور", 20, "480p"]
+        ["شروق الشمس على شاطئ البحر مع أمواج هادئة", 10, "720p"],
+        ["مدينة مستقبلية في الليل مع أضواء نيون", 15, "480p"],
+        ["حديقة جميلة مليئة بالزهور الملونة", 10, "480p"]
     ]
 )
 

requirements.txt

@@ -3,9 +3,10 @@ openai==0.27.0
 python-dotenv==0.19.0
 gradio==3.50.2
 torch==2.0.1
-diffusers==0.21.4
+diffusers==0.19.3
 transformers==4.30.2
 accelerate==0.21.0
+huggingface-hub==0.16.4
 pillow==9.5.0
 moviepy==1.0.3
 numpy==1.24.3