Update Gradio app with multiple files

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import openai
 import os
 import json
 import time
@@ -8,43 +7,15 @@ from typing import Optional, Tuple, Dict, Any
 import tempfile
 import base64
 import re
+from huggingface_hub import InferenceClient
 
-# Initialize OpenAI client with Poe API configuration
-client = openai.OpenAI(
-    api_key=os.getenv("POE_API_KEY"),  # Set your Poe API key as environment variable
-    base_url="https://api.poe.com/v1",
+# Initialize Hugging Face Inference Client with fal-ai provider
+client = InferenceClient(
+    provider="fal-ai",
+    api_key=os.environ["HF_TOKEN"],
+    bill_to="huggingface",
 )
 
-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 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,
@@ -52,59 +23,35 @@ def generate_video(
     api_key: Optional[str] = None
 ) -> Tuple[Optional[str], str]:
     """
-    Generate video using Sora-2 through Poe API.
+    Generate video using Sora-2 through Hugging Face Inference API with fal-ai provider.
     Returns tuple of (video_path, status_message).
     """
     try:
         # Use provided API key or environment variable
         if api_key:
-            temp_client = openai.OpenAI(
+            temp_client = InferenceClient(
+                provider="fal-ai",
                 api_key=api_key,
-                base_url="https://api.poe.com/v1",
+                bill_to="huggingface",
             )
         else:
             temp_client = client
-            if not os.getenv("POE_API_KEY") and not api_key:
-                return None, "❌ Please set POE_API_KEY environment variable."
-        
-        # Format prompt with parameters
-        formatted_prompt = format_sora_prompt(prompt, duration, size)
+            if not os.environ.get("HF_TOKEN") and not api_key:
+                return None, "❌ Please set HF_TOKEN environment variable."
         
-        # Call Sora-2 through Poe API
-        chat = temp_client.chat.completions.create(
-            model="Sora-2",
-            messages=[{"role": "user", "content": formatted_prompt}],
+        # Call Sora-2 through Hugging Face Inference API
+        video_bytes = temp_client.text_to_video(
+            prompt,
+            model="akhaliq/sora-2",
         )
         
-        # Extract the response content
-        content = chat.choices[0].message.content
+        # Save to temporary file
+        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_file:
+            tmp_file.write(video_bytes)
+            video_path = tmp_file.name
         
-        # Try to extract video URL from response
-        video_url = extract_video_url(content)
-        
-        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]}..."
+        status_message = f"✅ Video generated successfully!"
+        return video_path, status_message
         
     except Exception as e:
         error_msg = f"❌ Error generating video: {str(e)}"

requirements.txt

@@ -1,5 +1,5 @@
 gradio>=5.0.0
-openai>=1.0.0
 requests>=2.31.0
 numpy>=1.24.0
-Pillow>=10.0.0
+Pillow>=10.0.0
+huggingface-hub>=0.20.0