Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pandas as pd
 from huggingface_hub import login
 import re
+from groq import Groq
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -12,6 +13,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+        self.client = Groq(api_key=os.environ["GROQ_API_KEY"])
         self.agent_prompt = (
             """You are a general AI assistant. I will ask you a question. Report your thoughts, and
             finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
@@ -26,8 +28,7 @@ class BasicAgent:
         )
 
     def format_final_answer(self, answer: str) -> str:
-        """Ensure final answer matches required format"""
-        cleaned = " ".join(answer.split())  # Clean extra spaces and make sure it's a single line
+        cleaned = " ".join(answer.split())
         return f"FINAL ANSWER: {cleaned}"
 
     def check_commutativity(self):
@@ -59,7 +60,7 @@ class BasicAgent:
             }
         ) / len(words)
         return reversed_ratio > 0.3
-        
+
     def solve_riddle(self, question: str) -> str:
         question = question[::-1]
         if "opposite of the word" in question:
@@ -74,6 +75,22 @@ class BasicAgent:
                 return self.format_final_answer(opposite)
         return self.format_final_answer("COULD_NOT_SOLVE")
 
+    def query_groq(self, question: str) -> str:
+        full_prompt = f"{self.agent_prompt}\n\nQuestion: {question}"
+        try:
+            response = self.client.chat.completions.create(
+                model="llama3-8b-8192",
+                messages=[{"role": "user", "content": full_prompt}]
+            )
+            answer = response.choices[0].message.content
+            if "FINAL ANSWER:" in answer:
+                return answer.split("FINAL ANSWER:")[-1].strip()
+            else:
+                return self.format_final_answer(answer)
+        except Exception as e:
+            print(f"[Groq ERROR]: {e}")
+            return self.format_final_answer("GROQ_ERROR")
+
     def __call__(self, question: str) -> str:
         print(f"Received question: {question[:50]}...")
         if "commutative" in question.lower():
@@ -81,7 +98,7 @@ class BasicAgent:
         if self.maybe_reversed(question):
             print("Detected likely reversed riddle.")
             return self.solve_riddle(question)
-        return self.format_final_answer("NOT_A_RIDDLE")
+        return self.format_final_answer(self.query_groq(question))
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -168,4 +185,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)