Simplified model selections

app.py

@@ -1,60 +1,124 @@
-from fastapi import FastAPI
-from fastapi.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware
-import gradio as gr
+import asyncio
+import os
+from typing import Optional
+from gradio import Blocks, Markdown, Row, Column, Textbox, Audio, Button, Dropdown
+from dotenv import load_dotenv
 from scraper import scrape_url
 from podcast_generator import PodcastGenerator
 from tts import text_to_speech
-import uvicorn
-
-app = FastAPI()
-
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Mount static files
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
-def generate_podcast(url):
-    content = scrape_url(url)
-    podcast_generator = PodcastGenerator()
-    podcast_text = podcast_generator.generate_podcast(content)
-    audio_file = text_to_speech(podcast_text)
-    return audio_file
-
-# Create Gradio interface with custom configuration
-demo = gr.Interface(
-    fn=generate_podcast,
-    inputs=gr.Textbox(
-        label="Website URL",
-        placeholder="Enter the URL of the website you want to convert to a podcast"
-    ),
-    outputs=gr.Audio(label="Generated Podcast"),
-    title="URL to Podcast Generator",
-    description="Enter a URL to generate a podcast episode based on its content.",
-    theme="huggingface",
-    allow_flagging="never",
-    css="static/custom.css",  # Optional: for any custom CSS
-    analytics_enabled=False,
-)
-
-# Mount Gradio app with a specific base path
-app = gr.mount_gradio_app(app, demo, path="/")
-
-if __name__ == "__main__":
-    import os
-    # Create static directory if it doesn't exist
-    os.makedirs("static", exist_ok=True)
-    
-    uvicorn.run(
-        "app:app",
-        host="0.0.0.0",
-        port=7860,
-        reload=True
+from api_clients import OpenRouterClient, ElevenLabsClient
+
+load_dotenv()
+
+# Default choices for dropdowns
+default_voices = [("", "Enter API key to load voices")]
+default_models = [("", "Enter API key to load models")]
+
+async def generate_podcast(
+    url: str,
+    openrouter_key: str,
+    model_id: str,
+    elevenlabs_key: str,
+    voice_id: str
+) -> Optional[str]:
+    try:
+        content = scrape_url(url)
+        
+        # Initialize API clients
+        openrouter = OpenRouterClient(openrouter_key)
+        elevenlabs = ElevenLabsClient(elevenlabs_key)
+        
+        # Generate script using OpenRouter
+        script = await openrouter.generate_script(content, model_id)
+        
+        # Convert to audio using ElevenLabs
+        audio_file = elevenlabs.generate_audio(script, voice_id)
+        return audio_file
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def create_ui():
+    with Blocks(title='URL to Podcast Generator', theme='huggingface') as interface:
+        Markdown('# URL to Podcast Generator')
+        Markdown('Enter a URL to generate a podcast episode based on its content.')
+
+        with Row():
+            with Column(scale=2):
+                url_input = Textbox(
+                    label="Website URL",
+                    placeholder="Enter the URL of the website you want to convert to a podcast"
+                )
+                
+                with Row():
+                    with Column():
+                        openrouter_key = Textbox(
+                            label='OpenRouter API Key',
+                            type='password',
+                            placeholder='Enter key...'
+                        )
+                        openrouter_model = Dropdown(
+                            label='AI Model',
+                            choices=default_models,
+                            value=None,
+                            allow_custom_value=True
+                        )
+                        
+                    with Column():
+                        elevenlabs_key = Textbox(
+                            label='ElevenLabs API Key',
+                            type='password',
+                            placeholder='Enter key...'
+                        )
+                        voice_model = Dropdown(
+                            label='Voice',
+                            choices=default_voices,
+                            value=None,
+                            allow_custom_value=True
+                        )
+                
+                submit_btn = Button('Generate Podcast', variant='primary')
+
+            with Column(scale=1):
+                audio_output = Audio(label="Generated Podcast")
+                status = Textbox(label='Status', interactive=False)
+
+        # Event handlers for API key changes
+        async def update_openrouter_models(key):
+            if not key:
+                return gr.Dropdown(choices=default_models)
+            try:
+                client = OpenRouterClient(key)
+                models = await client.get_models()
+                return gr.Dropdown(choices=models)
+            except Exception as e:
+                return gr.Dropdown(choices=[(None, f"Error: {str(e)}")])
+
+        def update_elevenlabs_voices(key):
+            if not key:
+                return gr.Dropdown(choices=default_voices)
+            try:
+                client = ElevenLabsClient(key)
+                voices = client.get_voices()
+                return gr.Dropdown(choices=voices)
+            except Exception as e:
+                return gr.Dropdown(choices=[(None, f"Error: {str(e)}")])
+
+        # Set up event handlers
+        openrouter_key.change(fn=update_openrouter_models, inputs=openrouter_key, outputs=openrouter_model)
+        elevenlabs_key.change(fn=update_elevenlabs_voices, inputs=elevenlabs_key, outputs=voice_model)
+        
+        submit_btn.click(
+            fn=lambda *args: asyncio.run(generate_podcast(*args)),
+            inputs=[url_input, openrouter_key, openrouter_model, elevenlabs_key, voice_model],
+            outputs=[audio_output]
+        )
+
+    return interface
+
+if __name__ == '__main__':
+    demo = create_ui()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True
     )