Update ultimate_ai_app_final.py

ultimate_ai_app_final.py

@@ -1,91 +1,42 @@
-# ===========================
-# Ultimate AI App - Final Zero Interaction
-# ===========================
-
-# Required Libraries:
-# pip install diffusers transformers torch TTS moviepy pillow
-
-import torch
-from diffusers import StableDiffusionPipeline
-from TTS.api import TTS
-from transformers import pipeline
-from moviepy.editor import ImageSequenceClip
-import numpy as np
-from PIL import Image
-
-# ===========================
-# DEFAULT PROMPTS
-# ===========================
-DEFAULT_IMAGE_PROMPT = "3D, 4k, cinematic landscape, highly detailed"
-DEFAULT_VOICE_TEXT = "Hello! This is a fully automatic AI generation test."
-DEFAULT_TEXT_PROMPT = "Once upon a time in a futuristic world, AI created wonders."
-DEFAULT_VIDEO_PROMPT = "Fantasy landscape, cinematic, 4k"
-
-# ===========================
-# IMAGE GENERATION
-# ===========================
-def generate_image(prompt, save_path="generated_image.png"):
-    print("Generating image...")
-    pipe = StableDiffusionPipeline.from_pretrained(
-        "runwayml/stable-diffusion-v1-5",
-        torch_dtype=torch.float16
-    )
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipe = pipe.to(device)
-    image = pipe(prompt).images[0]
-    image.save(save_path)
-    print(f"✅ Image saved as {save_path}")
-
-# ===========================
-# VOICE GENERATION
-# ===========================
-def generate_voice(text, save_path="generated_voice.wav"):
-    print("Generating voice...")
-    tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=True, gpu=torch.cuda.is_available())
-    tts.tts_to_file(text=text, file_path=save_path)
-    print(f"✅ Voice saved as {save_path}")
-
-# ===========================
-# TEXT GENERATION
-# ===========================
-def generate_text(prompt, max_length=100):
-    print("Generating text...")
-    generator = pipeline('text-generation', model='gpt2')
-    result = generator(prompt, max_length=max_length, num_return_sequences=1)
-    with open("generated_text.txt", "w", encoding="utf-8") as f:
-        f.write(result[0]['generated_text'])
-    print(f"✅ Text saved as generated_text.txt")
-
-# ===========================
-# VIDEO GENERATION
-# ===========================
-def generate_video(prompt, num_frames=10, save_path="generated_video.mp4"):
-    print("Generating video...")
-    pipe = StableDiffusionPipeline.from_pretrained(
-        "runwayml/stable-diffusion-v1-5",
-        torch_dtype=torch.float16
-    )
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipe = pipe.to(device)
-
-    frames = []
-    for i in range(num_frames):
-        image = pipe(prompt).images[0]
-        frames.append(image.convert("RGB"))
-
-    clip = ImageSequenceClip([np.array(frame) for frame in frames], fps=2)
-    clip.write_videofile(save_path)
-    print(f"✅ Video saved as {save_path}")
-
-# ===========================
-# MAIN - ZERO INTERACTION
-# ===========================
-if __name__ == "__main__":
-    print("=== Ultimate AI App - Final Zero Interaction ===\n")
-
-    generate_image(DEFAULT_IMAGE_PROMPT)
-    generate_voice(DEFAULT_VOICE_TEXT)
-    generate_text(DEFAULT_TEXT_PROMPT)
-    generate_video(DEFAULT_VIDEO_PROMPT)
-
-    print("\n🎉 All tasks completed! All files are saved in this folder.")
+import gradio as gr
+from diffusers import StableDiffusionPipeline, DiffusionPipeline
+import torch, imageio
+
+# Load Image Generator
+img_model = "runwayml/stable-diffusion-v1-5"
+img_pipe = StableDiffusionPipeline.from_pretrained(img_model, torch_dtype=torch.float32)
+img_pipe = img_pipe.to("cpu")
+
+# Load Video Generator
+vid_model = "damo-vilab/text-to-video-ms-1.7b"
+vid_pipe = DiffusionPipeline.from_pretrained(vid_model, torch_dtype=torch.float32)
+vid_pipe = vid_pipe.to("cpu")
+
+# Image generation function
+def generate_image(prompt):
+    image = img_pipe(prompt).images[0]
+    return image
+
+# Video generation function
+def generate_video(prompt):
+    video_frames = vid_pipe(prompt, num_frames=8).frames
+    output_path = "output.mp4"
+    imageio.mimsave(output_path, video_frames, fps=8)
+    return output_path
+
+# Gradio Tabs UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎨 Free AI Image + Video Generator")
+    with gr.Tab("🖼️ Image Generator"):
+        text_in = gr.Textbox(label="ছবির জন্য প্রম্পট লিখুন")
+        img_out = gr.Image(type="pil")
+        btn1 = gr.Button("Generate Image")
+        btn1.click(generate_image, inputs=text_in, outputs=img_out)
+
+    with gr.Tab("🎬 Video Generator"):
+        text_in2 = gr.Textbox(label="ভিডিওর জন্য প্রম্পট লিখুন")
+        vid_out = gr.Video()
+        btn2 = gr.Button("Generate Video")
+        btn2.click(generate_video, inputs=text_in2, outputs=vid_out)
+
+demo.launch()