Update Gradio app with multiple files

app.py

@@ -7,6 +7,7 @@ import requests
 from typing import Optional, Tuple, Dict, Any
 import tempfile
 import base64
+import re
 
 # Initialize OpenAI client with Poe API configuration
 client = openai.OpenAI(
@@ -23,6 +24,27 @@ def format_sora_prompt(
     formatted_prompt = f"{prompt}\n\n--duration {duration} --size {size}"
     return formatted_prompt
 
+def extract_video_url(content: str) -> Optional[str]:
+    """Extract video URL from the response content."""
+    # Try to find direct video URL
+    url_pattern = r'https?://[^\s<>"{}|\\^`\[\]]+\.(?:mp4|mov|avi|webm)'
+    urls = re.findall(url_pattern, content)
+    if urls:
+        return urls[0]
+    
+    # Try to find any URL that might be a video
+    general_url_pattern = r'https?://[^\s<>"{}|\\^`\[\]]+'
+    urls = re.findall(general_url_pattern, content)
+    for url in urls:
+        if any(vid_keyword in url.lower() for vid_keyword in ['video', 'mp4', 'mov', 'media', 'cdn']):
+            return url
+    
+    # Return first URL if found
+    if urls:
+        return urls[0]
+    
+    return None
+
 def generate_video(
     prompt: str,
     duration: int = 8,
@@ -43,15 +65,12 @@ def generate_video(
         else:
             temp_client = client
             if not os.getenv("POE_API_KEY") and not api_key:
-                return None, "❌ Please provide a Poe API key or set POE_API_KEY environment variable."
+                return None, "❌ Please 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..."
-        
-        # 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,50 +79,48 @@ 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")
+        # Try to extract video URL from response
+        video_url = extract_video_url(content)
         
-        return video_path, status_message
+        if video_url:
+            # Download the video
+            try:
+                headers = {
+                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+                }
+                video_response = requests.get(video_url, headers=headers, timeout=60)
+                video_response.raise_for_status()
+                
+                # Save to temporary file
+                with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_file:
+                    tmp_file.write(video_response.content)
+                    video_path = tmp_file.name
+                
+                status_message = f"✅ Video generated successfully!"
+                return video_path, status_message
+                
+            except requests.exceptions.RequestException as e:
+                return None, f"❌ Failed to download video: {str(e)}"
+        else:
+            # If no URL found, the response might contain the video data directly
+            # or might need different parsing
+            return None, f"❌ Could not extract video from response. Response: {content[:200]}..."
         
     except Exception as e:
         error_msg = f"❌ Error generating video: {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
-
 def simple_generate(prompt: str) -> Optional[str]:
     """Simplified wrapper for generate_video that only takes prompt and returns video."""
-    video_path, _ = generate_video(prompt, duration=8, size="1280x720", api_key=None)
+    if not prompt or prompt.strip() == "":
+        return None
+    
+    video_path, status = generate_video(prompt, duration=8, size="1280x720", api_key=None)
+    
+    # If there's an error, you might want to handle it
+    if video_path is None:
+        print(f"Generation failed: {status}")
+    
     return video_path
 
 # Build the Gradio interface

requirements.txt

@@ -2,4 +2,13 @@ gradio>=5.0.0
 openai>=1.0.0
 requests>=2.31.0
 numpy>=1.24.0
-Pillow>=10.0.0
+Pillow>=10.0.0
+
+The main changes I made:
+1. Added `extract_video_url()` function to properly parse video URLs from the Poe API response
+2. Improved error handling to show what's in the response if video extraction fails
+3. Added proper headers for downloading videos
+4. Added validation for empty prompts
+5. Better error messages to help debug issues
+
+This should properly handle the Poe API response and extract/download the generated video.