Create app.py

app.py

@@ -0,0 +1,83 @@
+# gemini_video_gradio.py
+
+import gradio as gr
+import google.generativeai as genai
+import base64
+import requests
+import time
+import os
+
+# Configure your API key
+genai.configure(api_key=os.getenv("API_KEY") or "YOUR_API_KEY_HERE")
+
+def encode_image_to_base64(image_file):
+    if image_file is None:
+        return None
+    return base64.b64encode(image_file.read()).decode("utf-8")
+
+def generate_video(prompt, image_file):
+    image_bytes = encode_image_to_base64(image_file) if image_file else None
+
+    config = {
+        "model": "veo-2.0-generate-001",
+        "prompt": prompt,
+        "config": {
+            "numberOfVideos": 1,
+            # Optional settings:
+            # "durationSeconds": 3,
+            # "fps": 24,
+            # "resolution": "720p",
+        }
+    }
+
+    if image_bytes:
+        config["image"] = {
+            "imageBytes": image_bytes,
+            "mimeType": "image/png"
+        }
+
+    try:
+        op = genai.models.generate_videos(**config)
+        while not op.done:
+            time.sleep(1)
+            op = genai.operations.get_videos_operation(operation=op)
+
+        videos = op.response.get("generatedVideos", [])
+        if not videos:
+            return "No video was generated.", None
+
+        video_uri = videos[0]["video"]["uri"]
+        response = requests.get(video_uri)
+        filename = "generated_video.mp4"
+        with open(filename, "wb") as f:
+            f.write(response.content)
+
+        return "Video generated successfully!", filename
+
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
+# Build Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎥 Gemini Video Generator")
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Prompt", placeholder="e.g. A futuristic city at night")
+        image_input = gr.File(label="Optional Image (PNG/JPG)", file_types=[".png", ".jpg", ".jpeg"])
+
+    generate_btn = gr.Button("Generate Video")
+
+    status_output = gr.Textbox(label="Status")
+    video_output = gr.Video(label="Generated Video")
+
+    def run_generation(prompt, image):
+        return generate_video(prompt, image)
+
+    generate_btn.click(
+        fn=run_generation,
+        inputs=[prompt_input, image_input],
+        outputs=[status_output, video_output]
+    )
+
+# Run the app
+if __name__ == "__main__":
+    demo.launch()