Update app.py

app.py

@@ -11,42 +11,29 @@ load_dotenv()
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# CrewAI components (would normally be in separate files)
-class ResearchAgent:
-    def __init__(self):
-        self.role = "Researcher"
-        self.goal = "Research information thoroughly"
-    
-    def research(self, question):
-        # Implement research logic here
-        return f"Researched information about: {question}"
-
-class WritingAgent:
-    def __init__(self):
-        self.role = "Writer"
-        self.goal = "Write clear and accurate answers"
-    
-    def write(self, research_data):
-        # Implement writing logic here
-        return f"Comprehensive answer based on: {research_data}"
+# Import your CrewAI components (assuming they're in separate files)
+from agents import news_researcher, news_writer
+from tasks import research_task, write_task
 
-# Enhanced Agent Definition
 class CrewAIAgent:
     def __init__(self):
         print("Initializing CrewAI agents...")
-        self.researcher = ResearchAgent()
-        self.writer = WritingAgent()
+        self.crew = Crew(
+            agents=[news_researcher, news_writer],
+            tasks=[research_task, write_task],
+            process=Process.sequential,
+            verbose=True
+        )
         print("CrewAI agents initialized.")
 
     def __call__(self, question: str) -> str:
         print(f"Processing question: {question[:50]}...")
         
-        # Create and execute crew
         try:
-            research_data = self.researcher.research(question)
-            final_answer = self.writer.write(research_data)
-            print(f"Generated answer: {final_answer[:100]}...")
-            return final_answer
+            # Execute the crew with the question as input
+            result = self.crew.kickoff(inputs={'topic': question})
+            print(f"Generated answer: {result[:100]}...")
+            return str(result)
         except Exception as e:
             print(f"CrewAI error: {e}")
             return f"CrewAI Error: {str(e)}"
@@ -57,7 +44,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # Initialize agent
     try:
-        agent = CrewAIAgent()  # Using CrewAI instead of BasicAgent
+        agent = CrewAIAgent()
     except Exception as e:
         return f"Agent initialization failed: {e}", None