File size: 2,601 Bytes
7516245 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
import gradio as gr
from datetime import datetime
from core import generate_ai_news_blog
def create_gradio_interface():
"""
Create a Gradio interface for the AI News Blog Generator
Returns:
gr.Blocks: Gradio interface
"""
def run_generation(groq_key, tavily_key, selected_date):
if not groq_key or not tavily_key:
return "Please provide both API keys."
try:
result = generate_ai_news_blog(groq_key, tavily_key, selected_date)
return result
except Exception as e:
return f"Error generating blog: {str(e)}"
# Create the interface
with gr.Blocks(title="AI News Blog Generator") as demo:
gr.Markdown("# AI News Blog Generator")
gr.Markdown("Generate a daily roundup of AI news articles, categorized by topic.")
with gr.Row():
with gr.Column():
groq_key = gr.Textbox(label="Groq API Key", placeholder="Enter your Groq API key", type="password")
tavily_key = gr.Textbox(label="Tavily API Key", placeholder="Enter your Tavily API key", type="password")
date_picker = gr.Textbox(label="Date (YYYY-MM-DD)", placeholder="Leave empty for today's date",
value=datetime.now().strftime("%Y-%m-%d"))
with gr.Row():
generate_button = gr.Button("Generate AI News Blog", variant="primary")
clear_button = gr.Button("Clear Output")
with gr.Column():
status_text = gr.Textbox(label="Status", placeholder="Ready to generate", interactive=False)
output_md = gr.Markdown("Your AI News Blog will appear here.")
# Add loading state and status updates
generate_button.click(
fn=lambda: "Generating AI News Blog... This may take several minutes.",
inputs=None,
outputs=status_text,
queue=False
).then(
fn=run_generation,
inputs=[groq_key, tavily_key, date_picker],
outputs=output_md
).then(
fn=lambda: "Blog generation complete!",
inputs=None,
outputs=status_text
)
# Clear output
clear_button.click(
fn=lambda: ("Ready to generate", ""),
inputs=None,
outputs=[status_text, output_md]
)
return demo
def launch_ui():
"""Launch the Gradio web UI"""
demo = create_gradio_interface()
demo.launch() |