Update app.py

app.py

@@ -524,7 +524,7 @@ def create_blog_generator_workflow():
     
     return workflow.compile()
 
-def generate_ai_news_blog(groq_api_key=None, tavily_api_key=None, date=None):
+def generate_ai_news_blog(groq_api_key=None, tavily_api_key=None, date=None, topic=None):
     """Main function to generate AI news blog"""
     # Set API keys if provided
     if groq_api_key:
@@ -542,9 +542,14 @@ def generate_ai_news_blog(groq_api_key=None, tavily_api_key=None, date=None):
     else:
         today = date
     
+    # Create search query with date and topic to get recent news
+    query = f"latest artificial intelligence news {today} english"
+    if topic:
+        query = f"latest {topic} news {today} english"  # Update query to include user-defined topic
+    
     # Step 1: Search for AI news
     news_search = create_news_search_workflow()
-    news_results = news_search.invoke({"query": "latest artificial intelligence news", "date": today})
+    news_results = news_search.invoke({"query": query, "date": today})
     
     print(f"Found {len(news_results['news_items'])} AI news items")
     
@@ -592,12 +597,12 @@ def generate_ai_news_blog(groq_api_key=None, tavily_api_key=None, date=None):
 def create_gradio_interface():
     """Create a Gradio interface for the AI News Blog Generator"""
     
-    def run_generation(groq_key, tavily_key, selected_date):
+    def run_generation(groq_key, tavily_key, selected_date, topic):
         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)
+            result = generate_ai_news_blog(groq_key, tavily_key, selected_date, topic)  # Pass topic to the function
             return result
         except Exception as e:
             return f"Error generating blog: {str(e)}"
@@ -613,6 +618,7 @@ def create_gradio_interface():
                 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"))
+                topic_input = gr.Textbox(label="Topic", placeholder="Enter a topic for news generation")  # New input for topic
                 with gr.Row():
                     generate_button = gr.Button("Generate AI News Blog", variant="primary")
                     clear_button = gr.Button("Clear Output")
@@ -629,7 +635,7 @@ def create_gradio_interface():
             queue=False
         ).then(
             fn=run_generation,
-            inputs=[groq_key, tavily_key, date_picker],
+            inputs=[groq_key, tavily_key, date_picker, topic_input],  # Include topic in inputs
             outputs=output_md
         ).then(
             fn=lambda: "Blog generation complete!",