Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from workflow import ( | |
| MicroSiteGenerator, | |
| ) # Make sure workflow.py is in the same directory or accessible | |
| from agno.workflow import RunEvent | |
| import os | |
| from dotenv import load_dotenv | |
| import traceback # Import traceback for detailed error logging | |
| # Load environment variables from .env file, if present | |
| load_dotenv() | |
| # Instantiate the workflow | |
| # This will also initialize the agents defined in the workflow. | |
| microsite_workflow: MicroSiteGenerator = None | |
| try: | |
| microsite_workflow = MicroSiteGenerator() | |
| except Exception as e: | |
| print(f"Error initializing MicroSiteGenerator: {e}") | |
| traceback.print_exc() | |
| # Gradio UI will show a message if microsite_workflow is None | |
| def generate_microsite_app(audio_file_obj, audio_format_str, use_cache_bool): | |
| """ | |
| Gradio function to process audio and generate a microsite using the local workflow. | |
| audio_file_obj: Output from gr.Audio (type="filepath"). | |
| audio_format_str: The user-selected audio format. | |
| use_cache_bool: Boolean indicating whether to use the transcription cache. | |
| """ | |
| if microsite_workflow is None: | |
| return ( | |
| "Critical Error: Workflow failed to initialize. Check console logs.", | |
| "App is not functional. Please ensure all configurations (like API keys) are correctly set.", | |
| ) | |
| if audio_file_obj is None: | |
| return "Status: Idle", "Please upload an audio file to begin. 🎤" | |
| audio_source_path = audio_file_obj # This is already the path string | |
| processing_log_entries = ["🚀 Starting microsite generation..."] | |
| final_result_markdown = "⏳ Processing... please wait." | |
| try: | |
| # The run method is a generator. We iterate to get the final result. | |
| for response in microsite_workflow.run( | |
| audio_source=audio_source_path, | |
| audio_format=audio_format_str.lower(), # Ensure format is lowercase | |
| ): | |
| processing_log_entries.append(f"🔄 Workflow event: {response.event.value}") | |
| if response.event == RunEvent.workflow_completed: | |
| content = response.content # This is site_details from the workflow | |
| if isinstance(content, dict): # Expected site_details dictionary | |
| if content.get("success"): | |
| site_url = content.get("site", {}).get("url") | |
| site_name = content.get("site", {}).get("name", "N/A") | |
| admin_url = content.get("site", {}).get("admin_url", "#") | |
| if site_url: | |
| final_result_markdown = ( | |
| f"🎉 **Microsite '{site_name}' Deployed!** 🎉\n\n" | |
| f"🔗 **Access it here:** [{site_url}]({site_url})\n\n" | |
| f"<details><summary>ℹ️ Deployment Details (Admin Link)</summary>" | |
| f"<p>Admin URL: <a href='{admin_url}' target='_blank' rel='noopener noreferrer'>{admin_url}</a></p>" | |
| f"</details>" | |
| ) | |
| processing_log_entries.append("✅ Deployment successful.") | |
| else: | |
| final_result_markdown = "⚠️ Microsite generated, but deployment URL was not found in the response." | |
| processing_log_entries.append("❌ Deployment URL missing.") | |
| else: # Deployment failed or error reported by deploy_html_file_with_digest | |
| error_msg = content.get("message", "Unknown deployment error.") | |
| final_result_markdown = ( | |
| f"❌ **Microsite Generation Failed:** {error_msg}" | |
| ) | |
| processing_log_entries.append( | |
| f"❌ Deployment/Generation failed: {error_msg}" | |
| ) | |
| elif isinstance(content, str) and "Site was not generated" in content: # Workflow specific error string | |
| final_result_markdown = f"❌ **Site Not Generated:** {content}. This often indicates a transcription failure or issue with the audio." | |
| processing_log_entries.append( | |
| f"❌ Site generation aborted: {content}" | |
| ) | |
| else: | |
| final_result_markdown = f"🤔 Workflow completed with unexpected content: {str(content)[:200]}..." # Truncate long content | |
| processing_log_entries.append( | |
| f"⚠️ Unexpected content: {str(content)[:200]}..." | |
| ) | |
| break # Critical final event processed | |
| else: | |
| # Handle other potential intermediate events if the workflow is updated to yield more | |
| # For example, if the workflow yields progress updates: | |
| if hasattr(response.content, 'get') and response.content.get('progress_message'): | |
| processing_log_entries.append(f"⏳ {response.content['progress_message']}") | |
| elif isinstance(response.content, str): | |
| processing_log_entries.append(f"ℹ️ {response.content[:100]}...") | |
| except Exception as e: | |
| tb_str = traceback.format_exc() | |
| processing_log_entries.append( | |
| f"💥 Critical error during workflow execution: {str(e)}" | |
| ) | |
| print(f"Error in generate_microsite_app: {e}\n{tb_str}") # Log to console for debugging | |
| final_result_markdown = ( | |
| f"💥 **An Unexpected Error Occurred!** 💥\n\n" | |
| f"Details: `{str(e)}`\n\n" | |
| f"Please check the console logs for more information or try again. " | |
| f"If the problem persists, ensure all configurations and API keys are correctly set." | |
| ) | |
| return "\n".join(processing_log_entries), final_result_markdown | |
| # Define common audio formats | |
| COMMON_AUDIO_FORMATS = [ | |
| "wav", "mp3", "m4a", "flac", "ogg", "aac", "opus", "amr", "webm", | |
| ] | |
| # Prepare environment variable warning message for the UI | |
| missing_env_vars_messages = [] | |
| if not os.getenv("NETLIFY_PERSONAL_ACCESS_TOKEN"): | |
| missing_env_vars_messages.append("`NETLIFY_PERSONAL_ACCESS_TOKEN` (for deploying the site to Netlify)") | |
| if not os.getenv("GOOGLE_API_KEY"): # Common for Gemini models | |
| missing_env_vars_messages.append("`GOOGLE_API_KEY` (for AI models like Google Gemini)") | |
| # Add other critical env vars checks here if your workflow agents need them | |
| env_warning_html = "" | |
| if missing_env_vars_messages: | |
| vars_list_html = "".join([f"<li>{var}</li>" for var in missing_env_vars_messages]) | |
| env_warning_html = ( | |
| f"<div style='background-color: #332200; color: #FFDDAA; border: 1px solid #553300; padding: 15px; margin-bottom:20px; border-radius: 5px;'>" | |
| f"<strong>⚠️ Heads up!</strong> The application might be missing the following environment variable(s):" | |
| f"<ul style='margin-top: 10px; margin-bottom: 0; padding-left: 20px;'>{vars_list_html}</ul>" | |
| f"This could affect its functionality. Please ensure they are set in your environment or `.env` file." | |
| f"</div>" | |
| ) | |
| # Workflow description for the UI | |
| workflow_desc_html = "" | |
| if microsite_workflow and hasattr(microsite_workflow, "description"): | |
| escaped_description = microsite_workflow.description.replace("\n", "<br>") | |
| workflow_desc_html = f""" | |
| <details style="margin-top:15px; margin-bottom:15px; padding:10px; background-color:#2a2a2a; border-radius:5px; border: 1px solid #444;"> | |
| <summary style="font-weight:bold; cursor:pointer;">📖 Click to see Workflow Details</summary> | |
| <p style="margin-top:10px;"><em>{escaped_description}</em></p> | |
| </details> | |
| """ | |
| elif microsite_workflow is None: | |
| workflow_desc_html = "<p style='color:red; font-weight:bold;'>WORKFLOW INITIALIZATION FAILED. Please check console logs for errors. API keys might be missing or other configurations might be incorrect.</p>" | |
| else: | |
| workflow_desc_html = "<p style='color:orange;'>Workflow description not available.</p>" | |
| app_title = "MicrositePilot 🎙️➡️🌐" | |
| app_intro_markdown = f""" | |
| {env_warning_html} | |
| Welcome to **MicrositePilot**! Upload a product demo call recording (audio file). | |
| The AI will transcribe it, extract key information, and generate a personalized recap microsite, automatically deployed to Netlify. | |
| {workflow_desc_html} | |
| """ | |
| custom_css = """ | |
| body { font-family: 'Inter', sans-serif; } | |
| .gradio-container { max-width: 900px !important; margin: auto !important; } | |
| footer { display: none !important; } /* Hide default Gradio footer */ | |
| h1 { text-align: center; } | |
| .gr-button { box-shadow: 0 1px 3px 0 rgba(0,0,0,.1), 0 1px 2px 0 rgba(0,0,0,.06); } | |
| """ | |
| with gr.Blocks(theme="dark", css=custom_css) as demo: | |
| gr.Markdown(f"<h1>{app_title}</h1>") | |
| gr.HTML(app_intro_markdown) | |
| with gr.Row(equal_height=False): | |
| with gr.Column(scale=1, min_width=300): | |
| gr.Markdown("### 📤 Step 1: Upload Audio") | |
| audio_input = gr.Audio( | |
| type="filepath", | |
| label="Product Demo Audio File (WAV, MP3, M4A, etc.)", | |
| ) | |
| gr.Markdown("### ⚙️ Step 2: Configure Options") | |
| audio_format_input = gr.Dropdown( | |
| choices=COMMON_AUDIO_FORMATS, | |
| label="Original Audio Format (Crucial)", | |
| value="mp3", # Default common format | |
| ) | |
| cache_checkbox = gr.Checkbox( | |
| label="Use Transcription Cache ⚡ (Speeds up re-runs)", | |
| value=True, # Default to using cache | |
| ) | |
| gr.Markdown("### ✨ Step 3: Generate!") | |
| submit_button = gr.Button( | |
| "Generate Microsite", variant="primary", elem_id="submit_button_custom" | |
| ) | |
| with gr.Column(scale=2, min_width=400): | |
| gr.Markdown("### 📊 Results") | |
| log_output = gr.Textbox( | |
| label="⚙️ Processing Log", | |
| lines=12, | |
| interactive=False, | |
| placeholder="Workflow updates and logs will appear here...", | |
| ) | |
| microsite_link_output = gr.Markdown( | |
| label="🔗 Microsite Output", | |
| value="Your deployed microsite link and details will appear here once generated.", | |
| ) | |
| example_audio_file = "Listen to an A.I. sales rep cold call (and close) a prospect. #ai #sales.mp3" | |
| if os.path.exists(example_audio_file): | |
| gr.Examples( | |
| examples=[[example_audio_file, "mp3", True]], | |
| inputs=[audio_input, audio_format_input, cache_checkbox], | |
| outputs=[log_output, microsite_link_output], | |
| fn=generate_microsite_app, | |
| cache_examples=False, | |
| label="📋 Example (click to run)", | |
| ) | |
| else: | |
| gr.Markdown( | |
| "<p style='text-align:center; font-style:italic; color:grey;'>Note: Example audio file 'Listen to an A.I. sales rep cold call (and close) a prospect. #ai #sales.mp3' not found. Examples disabled.</p>" | |
| ) | |
| submit_button.click( | |
| fn=generate_microsite_app, | |
| inputs=[audio_input, audio_format_input, cache_checkbox], | |
| outputs=[log_output, microsite_link_output], | |
| api_name="generate_microsite", | |
| ) | |
| if __name__ == "__main__": | |
| if microsite_workflow is None: | |
| print("CRITICAL: MicroSiteGenerator workflow failed to initialize. The Gradio app might not function correctly.") | |
| print("Please check for errors above, ensure API keys (e.g., GOOGLE_API_KEY, NETLIFY_PERSONAL_ACCESS_TOKEN) are set in your .env file or environment, and all dependencies are installed.") | |
| else: | |
| print("MicroSiteGenerator workflow initialized successfully.") | |
| if not os.getenv("NETLIFY_PERSONAL_ACCESS_TOKEN"): | |
| print("CONSOLE REMINDER: NETLIFY_PERSONAL_ACCESS_TOKEN is not set. Deployment to Netlify will fail.") | |
| if not os.getenv("GOOGLE_API_KEY"): | |
| print("CONSOLE REMINDER: GOOGLE_API_KEY is not set. AI agent calls may fail.") | |
| print("Gradio app starting...") | |
| demo.launch() | |