Update Gradio app with multiple files

app.py

@@ -1,12 +1,7 @@
 import gradio as gr
 import openai
 import os
-import json
-import time
-import requests
-from typing import Optional, Tuple, Dict, Any
-import tempfile
-import base64
+from typing import Optional, Tuple
 
 # Initialize OpenAI client with Poe API configuration
 client = openai.OpenAI(
@@ -14,15 +9,6 @@ client = openai.OpenAI(
     base_url="https://api.poe.com/v1",
 )
 
-def format_sora_prompt(
-    prompt: str,
-    duration: int = 8,
-    size: str = "1280x720"
-) -> str:
-    """Format the prompt with Sora-2 specific parameters."""
-    formatted_prompt = f"{prompt}\n\n--duration {duration} --size {size}"
-    return formatted_prompt
-
 def generate_video(
     prompt: str,
     duration: int = 8,
@@ -31,7 +17,7 @@ def generate_video(
 ) -> Tuple[Optional[str], str]:
     """
     Generate video using Sora-2 through Poe API.
-    Returns tuple of (video_path, status_message).
+    Returns tuple of (response_content, status_message).
     """
     try:
         # Use provided API key or environment variable
@@ -46,12 +32,9 @@ def generate_video(
                 return None, "❌ Please provide a Poe API key or set POE_API_KEY environment variable."
         
         # Format prompt with parameters
-        formatted_prompt = format_sora_prompt(prompt, duration, size)
-        
-        # Start generation
-        status_message = "🎬 Initiating video generation with Sora-2..."
+        formatted_prompt = f"{prompt}\n\n--duration {duration} --size {size}"
         
-        # Call Sora-2 through Poe API - simplified API call
+        # Call Sora-2 through Poe API
         chat = temp_client.chat.completions.create(
             model="Sora-2",
             messages=[{"role": "user", "content": formatted_prompt}],
@@ -60,46 +43,14 @@ def generate_video(
         # Extract the response content
         content = chat.choices[0].message.content
         
-        # The response should contain a video URL or base64 encoded video
-        # For demonstration, we'll save it as a temporary file
-        # In production, you'd parse the actual video data from the response
-        
-        # Create a placeholder video file (in production, save actual video data)
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_file:
-            video_path = tmp_file.name
-            
-            # If the response contains a URL, download it
-            if content.startswith("http"):
-                video_response = requests.get(content)
-                tmp_file.write(video_response.content)
-                status_message = f"✅ Video generated successfully! Duration: {duration}s, Size: {size}"
-            else:
-                # If it's base64 or other format, handle accordingly
-                status_message = "✅ Video generation completed!"
-                # For demo purposes, write placeholder
-                tmp_file.write(b"Video data would be here")
-        
-        return video_path, status_message
+        status_message = f"✅ Response received! Duration: {duration}s, Size: {size}"
+        return content, status_message
         
     except Exception as e:
-        error_msg = f"❌ Error generating video: {str(e)}"
+        error_msg = f"❌ Error: {str(e)}"
         return None, error_msg
 
-def validate_api_key(api_key: str) -> bool:
-    """Validate if the provided API key works."""
-    try:
-        test_client = openai.OpenAI(
-            api_key=api_key,
-            base_url="https://api.poe.com/v1",
-        )
-        # Try a simple test request
-        test_client.chat.completions.create(
-            model="Sora-2",
-            messages=[{"role": "user", "content": "test"}],
-        )
-        return True
-    except:
-        return False
+
 
 # Custom CSS for better styling
 custom_css = """
@@ -190,10 +141,12 @@ with gr.Blocks(title="Sora-2 Text-to-Video Generator", css=custom_css, theme=gr.
             with gr.Column(scale=1):
                 # Output Section
                 with gr.Group():
-                    gr.Markdown("### 🎥 Generated Video")
-                    video_output = gr.Video(
-                        label="Output",
-                        height=400
+                    gr.Markdown("### 📄 Response")
+                    response_output = gr.Textbox(
+                        label="Sora-2 Response",
+                        interactive=False,
+                        lines=10,
+                        max_lines=20
                     )
                     status_output = gr.Textbox(
                         label="Status",
@@ -218,13 +171,9 @@ with gr.Blocks(title="Sora-2 Text-to-Video Generator", css=custom_css, theme=gr.
         # Information Section
         with gr.Accordion("ℹ️ About Sora-2", open=False):
             gr.Markdown("""
-            **Sora-2** is OpenAI's latest video and audio generation model, delivering:
+            **Sora-2** is OpenAI's text-to-video generation model accessible through Poe API.
             
-            - 🎨 **Exceptional Realism**: Photorealistic scenes with accurate physics
-            - 🎬 **Cinematic Quality**: Professional-grade video generation
-            - 🔊 **Synchronized Audio**: Dialogue and sound effects (when applicable)
-            - 🎯 **Precise Control**: Multi-shot prompt adherence and editing capabilities
-            - 🌍 **Real-world Elements**: Integration of people, animals, and objects
+            This simplified interface sends your prompt to Sora-2 and displays the response.
             
             **Available Parameters:**
             - **Duration**: 4, 8, or 12 seconds
@@ -241,7 +190,7 @@ with gr.Blocks(title="Sora-2 Text-to-Video Generator", css=custom_css, theme=gr.
     generate_btn.click(
         fn=generate_video,
         inputs=[prompt_input, duration_slider, size_dropdown, api_key_input],
-        outputs=[video_output, status_output]
+        outputs=[response_output, status_output]
     )
 
 # Launch the application