Update app.py

app.py

@@ -1,77 +1,47 @@
 import os
-import google-genai as genai
-import gradio as gr
-import base64
 import time
-import requests
-
-# Set up your API key
-genai.configure(api_key=os.getenv("GEMINI_API_KEY", "your-api-key-here"))
-
-# Helper to encode image
-def encode_image(image_path):
-    with open(image_path, "rb") as f:
-        return base64.b64encode(f.read()).decode("utf-8")
-
-# Polling for operation
-def poll_operation(op_name):
-    while True:
-        op = genai.get_operation(name=op_name)
-        if op.get("done"):
-            return op
-        time.sleep(2)
-
-# Main generation function
-def generate_video(prompt, image):
-    if not prompt or image is None:
-        return "Prompt and image are required.", None
-
-    image_b64 = encode_image(image)
-
-    model = genai.GenerativeModel("veo-2.0-generate-001")  # Use correct model name
-    try:
-        op = model.generate_videos(
-            prompt=prompt,
-            image={
-                "imageBytes": image_b64,
-                "mimeType": "image/png"
-            },
-            config={
-                "numberOfVideos": 1
-            }
-        )
-
-        op_result = poll_operation(op["name"])
-        videos = op_result["response"].get("generatedVideos", [])
-
-        if not videos:
-            return "No video returned", None
-
-        video_url = videos[0]["video"]["uri"]
-        video_data = requests.get(video_url).content
-
-        output_path = "generated_video.mp4"
-        with open(output_path, "wb") as f:
-            f.write(video_data)
-
-        return "Video generated successfully.", output_path
-
-    except Exception as e:
-        return f"Error: {str(e)}", None
+import gradio as gr
+from google import genai
+
+# Initialize client with API key from env var
+client = genai.Client()
+
+def generate_video(prompt):
+    operation = client.models.generate_videos(
+        model="veo-3.0-generate-preview",
+        prompt=prompt,
+    )
+
+    while not operation.done:
+        print("Waiting for video generation to complete...")
+        time.sleep(10)
+        operation = client.operations.get(operation)
+
+    generated_video = operation.response.generated_videos[0]
+    # Download video bytes
+    video_file = client.files.download(file=generated_video.video)
+    
+    # Save locally
+    video_path = "dialogue_example.mp4"
+    with open(video_path, "wb") as f:
+        f.write(video_file)
+    
+    return video_path
 
-# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎬 Gemini Veo Video Generator")
-    prompt_input = gr.Textbox(label="Prompt")
-    image_input = gr.Image(type="filepath", label="Upload an Image")
+    gr.Markdown("# Gemini Veo Video Generator")
+    prompt_input = gr.Textbox(label="Video Prompt", value="""A close up of two people staring at a cryptic drawing on a wall, torchlight flickering.
+A man murmurs, 'This must be it. That's the secret code.' The woman looks at him and whispering excitedly, 'What did you find?'""")
     generate_btn = gr.Button("Generate Video")
-    status = gr.Textbox(label="Status")
     video_output = gr.Video(label="Generated Video")
+    status = gr.Textbox(label="Status")
 
-    def run(prompt, image):
-        msg, path = generate_video(prompt, image)
-        return msg, path
+    def on_generate(prompt):
+        status.update("Generating video, please wait...")
+        video_path = generate_video(prompt)
+        status.update("Generation complete!")
+        return video_path, "Generation complete!"
 
-    generate_btn.click(fn=run, inputs=[prompt_input, image_input], outputs=[status, video_output])
+    generate_btn.click(on_generate, inputs=prompt_input, outputs=[video_output, status])
 
 demo.launch()