Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import requests
 from bs4 import BeautifulSoup
 import trafilatura
-from smolagents import Agent
+from smolagents import create_agent
 
 # Streamlit UI
 def main():
@@ -51,13 +51,22 @@ def find_relevant_article(base_url, question):
 
 # Step 5: Generate Answer using `smolagents`
 def generate_answer(question, context):
-    agent = Agent("Question-Answering Agent", description="Answers questions based on documentation.")
-    prompt = f"""
-    Context: {context}
-    Question: {question}
-    Provide a clear and concise answer.
-    """
-    return agent.run(prompt)
+    """Defines an AI agent to generate answers based on documentation context."""
+    
+    def answer_logic(state):
+        """Agent logic to answer based on context."""
+        return f"Based on the documentation, here is my answer: {state['context'][:500]}..."  # Truncating for brevity
+
+    # Create the agent
+    agent = create_agent(
+        name="QA_Agent",
+        description="Answers questions based on documentation content.",
+        process=answer_logic,
+    )
+
+    # Run the agent
+    response = agent({"context": context, "question": question})
+    return response
 
 if __name__ == "__main__":
-    main()
+    main()