import os
import gradio as gr
import torch
from diffusers import StableDiffusionPipeline

# মডেল আইডি এবং পাইপলাইন সেটআপ (১৬ জিবি র‍্যাম ও CPU-এর জন্য অপ্টিমাইজড)
model_id = "OFA-Sys/small-stable-diffusion-v0"

# torch_dtype এর জায়গায় সঠিক নিয়ম (dtype) ব্যবহার করা হলো অবসোলেট এরর এড়াতে
pipe = StableDiffusionPipeline.from_pretrained(model_id, dtype=torch.float32)
pipe = pipe.to("cpu")

def generate_video_frame(prompt):
    # প্রম্পট দিয়ে ইমেজ ফ্রেম তৈরি
    image = pipe(prompt, num_inference_steps=15).images[0]
    return image

# গ্রাডিও ইউজার ইন্টারফেস (মোবাইল ফ্রেন্ডলি)
interface = gr.Interface(
    fn=generate_video_frame,
    inputs=gr.Textbox(label="আপনার এআই প্রম্পট লিখুন", placeholder="A running cat, cinematic..."),
    outputs=gr.Image(label="Generated Output"),
    title="Rayhan's Free AI Space",
    description="এই স্পেসটি সম্পূর্ণ ফ্রিতে এবং আনলিমিটেড বার ব্যবহার করতে পারবেন।"
)

if __name__ == "__main__":
    # ভুল 'docker_setting' আর্গুমেন্টটি ফেলে দিয়ে ক্লিন কোড করা হলো
    interface.launch(server_name="0.0.0.0", server_port=7860)