Update app.py

app.py

@@ -8,23 +8,32 @@ import gradio as gr
 import os
 import ast
 
+# Load environment variables
 load_dotenv()
 google_api_key = os.getenv("GOOGLE_API_KEY")
+user_agent = os.getenv("USER_AGENT", "Caramel AI Bot")
 model = "gemini-2.5-flash"
 
 @tool
 def web_scrape_tool(urls: str) -> str:
+    """Scrapes content from a list of URLs.
+    The input should be a string representation of a Python list of URLs
+    (e.g., "['https://hereandnowai.com', 'https://hereandnow.co.in']").
+    Returns the concatenated text content of all scraped pages."""
     try:
         url_list = ast.literal_eval(urls)
         if not isinstance(url_list, list) or not all(isinstance(url, str) for url in url_list):
             return "Invalid input format. Please provide a list of URLs as a string."
-    except Exception:
+    except (ValueError, SyntaxError):
         return "Invalid input format. Please provide a list of URLs as a string."
-    
+
     combined_content = []
     for url in url_list:
         try:
-            loader = WebBaseLoader([url], requests_kwargs={"headers": {"User-Agent": "Caramel AI"}})
+            loader = WebBaseLoader(
+                [url],
+                requests_kwargs={"headers": {"User-Agent": user_agent}},
+            )
             documents = loader.load()
             for doc in documents:
                 combined_content.append(doc.page_content)
@@ -32,24 +41,26 @@ def web_scrape_tool(urls: str) -> str:
             combined_content.append(f"Could not scrape {url}. Error: {e}")
     return "\n\n".join(combined_content)
 
-# Preload agent only once (efficient)
+# Setup the LLM and Agent
 llm = ChatGoogleGenerativeAI(model=model, google_api_key=google_api_key)
 tools = [web_scrape_tool]
 prompt = hub.pull("hwchase17/react")
-agent = create_react_agent(llm, tools, prompt)
+agent = create_react_agent(llm=llm, tools=tools, prompt=prompt)
 agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=False, handle_parsing_errors=True)
 
-# Gradio interface
-def ask_agent(user_question):
+# Gradio interface function
+def ask_agent(user_input):
     try:
-        result = agent_executor.invoke({"input": user_question})
-        return result['output']
+        result = agent_executor.invoke({"input": user_input})
+        return result["output"]
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {e}"
 
+# Launch Gradio app
 gr.Interface(
     fn=ask_agent,
-    inputs=gr.Textbox(label="Ask a question (include the URL)"),
-    outputs=gr.Textbox(label="Agent's Response"),
-    title="Web Scraping Agent with Gemini"
+    inputs=gr.Textbox(lines=2, placeholder="Ask a question with a URL...", label="Your Query"),
+    outputs=gr.Textbox(lines=10, label="Agent's Response"),
+    title="Gemini Web Scraping Agent",
+    description="Ask a question about one or more web pages. Include URLs in the question."
 ).launch()