Update app.py

app.py

@@ -25,27 +25,32 @@ web_agent = Agent(
     markdown=True  # Format the output in Markdown
 )
 
-# Function to fetch the latest AI news
-def fetch_latest_ai_news():
-    query = "Find the latest news about Artificial Intelligence and summarize it."
-    response = web_agent.execute(query)  # Fetch and summarize AI news using 'execute' method
+# Function to fetch news based on user query
+def fetch_news(query):
+    response = web_agent.execute(query)  # Fetch and summarize news using 'execute' method
     return response
 
 # Streamlit App
 st.title("AI News Fetcher")
 
 st.markdown("""
-Welcome to the **AI News Fetcher** app! This tool provides the latest summaries of AI-related news from the web.
+Welcome to the **AI News Fetcher** app! Enter your query to find the latest summaries of AI-related news or any topic of interest.
 
 ---
 """)
 
-if st.button("Fetch Latest AI News"):
-    st.info("Fetching the latest AI news. This may take a few seconds...")
-    try:
-        news_summary = fetch_latest_ai_news()
-        st.markdown(news_summary)  # Display the news summary in Markdown format
-    except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+# User input for query
+user_query = st.text_input("Enter your query:", placeholder="Find the latest news about AI, summarize advancements in robotics, etc.")
+
+if st.button("Fetch News"):
+    if user_query.strip():
+        st.info("Fetching the news. This may take a few seconds...")
+        try:
+            news_summary = fetch_news(user_query)
+            st.markdown(news_summary)  # Display the news summary in Markdown format
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+    else:
+        st.warning("Please enter a valid query.")
 else:
-    st.write("Click the button above to fetch the latest AI news.")
+    st.write("Enter a query and click the button to fetch news.")