Create app.py

app.py

@@ -0,0 +1,83 @@
+import gradio as gr
+import subprocess
+import json
+import time
+
+def generate_video(prompt, aspect_ratio):
+    # Prepare the API key and base URL
+    BASE_URL = "https://generativelanguage.googleapis.com/v1beta"
+    API_KEY = "AIzaSyDljfPA5gl7zn11UamTPllqdX13-q85ZgY"  # Note: You should use environment variables for the API key
+    
+    # Prepare the request data
+    request_data = {
+        "instances": [{
+            "prompt": prompt
+        }],
+        "parameters": {
+            "aspectRatio": aspect_ratio,
+            "personGeneration": "dont_allow",
+        }
+    }
+    
+    try:
+        # Prepare the curl command
+        curl_command = [
+            "curl",
+            f"{BASE_URL}/models/veo-2.0-generate-001:predictLongRunning?key={API_KEY}",
+            "-H", "Content-Type: application/json",
+            "-X", "POST",
+            "-d", json.dumps(request_data)
+        ]
+        
+        # Execute the curl command
+        result = subprocess.run(curl_command, capture_output=True, text=True)
+        
+        if result.returncode != 0:
+            return f"Error: {result.stderr}"
+        
+        # Parse the response
+        response = json.loads(result.stdout)
+        operation_name = response.get("name", "")
+        
+        # Return the operation name and full response
+        return f"Operation Name: {operation_name}\n\nFull Response:\n{json.dumps(response, indent=2)}"
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
+# Create the Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Veo Video Generation API")
+    gr.Markdown("Enter your prompt and select aspect ratio to generate a video using Google's Veo API.")
+    
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(label="Video Prompt", placeholder="Describe the video you want to generate...")
+            aspect_ratio = gr.Dropdown(
+                label="Aspect Ratio",
+                choices=["16:9", "9:16", "1:1", "4:3", "3:4"],
+                value="16:9"
+            )
+            submit_btn = gr.Button("Generate Video")
+        
+        with gr.Column():
+            output = gr.Textbox(label="API Response", interactive=False, lines=10)
+    
+    submit_btn.click(
+        fn=generate_video,
+        inputs=[prompt, aspect_ratio],
+        outputs=output
+    )
+    
+    gr.Examples(
+        examples=[
+            ["Panning wide shot of a calico kitten sleeping in the sunshine", "16:9"],
+            ["Aerial view of a tropical beach at sunset", "16:9"],
+            ["Close-up of a chef preparing sushi in a restaurant kitchen", "9:16"]
+        ],
+        inputs=[prompt, aspect_ratio]
+    )
+
+# Launch the interface
+if __name__ == "__main__":
+    demo.launch()