Update app.py

app.py

@@ -22,13 +22,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        api_token = os.getenv("HF_TOKEN")
-        
-        os.environ["HUGGINGFACEHUB_API_TOKEN"] = "hf_abcdefghijklmnopqrstuvwxyz"
-        login(api_token)
-        print(f"Using API Token: {api_token}")
-        self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
-        self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")  # Initialize tokenizer
         SYSTEM_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].
         YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated
@@ -44,7 +37,12 @@ class BasicAgent:
 
     def maybe_reversed(self, text: str) -> bool:
         words = text.split()
-        reversed_ratio = sum(1 for word in words if word[::-1].lower() in {"if", "you", "understand", "this", "sentence", "write", "opposite", "of", "the", "word", "left", "answer"}) / len(words)
+        reversed_ratio = sum(
+            1 for word in words if word[::-1].lower() in {
+                "if", "you", "understand", "this", "sentence", "write",
+                "opposite", "of", "the", "word", "left", "answer"
+            }
+        ) / len(words)
         return reversed_ratio > 0.3
 
     def solve_riddle(self, question: str) -> str:
@@ -53,60 +51,30 @@ class BasicAgent:
             match = re.search(r"opposite of the word ['\"](\w+)['\"]", reversed_text)
             if match:
                 word = match.group(1).lower()
-                opposites = {"left": "right", "up": "down", "hot": "cold", "true": "false", "yes": "no", "black": "white"}
+                opposites = {
+                    "left": "right", "up": "down", "hot": "cold",
+                    "true": "false", "yes": "no", "black": "white"
+                }
                 opposite = opposites.get(word, f"UNKNOWN_OPPOSITE_OF_{word}")
                 return f"FINAL ANSWER: {opposite}"
-        return f"FINAL ANSWER: COULD_NOT_SOLVE"
-
-    def is_chess_question(self, question: str) -> bool:
-        keywords = ["chess", "algebraic notation", "checkmate", "bishop", "rook", "knight", "position", "board"]
-        return any(k in question.lower() for k in keywords)
-
-    def extract_board_from_image(self, image_data: str) -> chess.Board:
-        # Stub implementation; replace with real image-to-FEN logic
-        # For now, simulate board with forced mate
-        board = chess.Board()
-        board.set_fen("6k1/5ppp/8/8/8/8/5PPP/6K1 b - - 0 1")  # Example stub position
-        return board
-
-    def find_best_move(self, board: chess.Board) -> str:
-        try:
-            with chess.engine.SimpleEngine.popen_uci("/usr/games/stockfish") as engine:
-                result = engine.play(board, chess.engine.Limit(time=0.1))
-                return f"FINAL ANSWER: {board.san(result.move)}"
-        except Exception as e:
-            return f"FINAL ANSWER: ENGINE_ERROR: {e}"
+        return "FINAL ANSWER: COULD_NOT_SOLVE"
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-
-        # Tokenize input question using tokenizer
-        inputs = self.tokenizer(question, return_tensors="pt", truncation=True, padding=True, max_length=512)
-
-        if self.is_chess_question(question):
-            print("Detected chess-related question. Attempting board extraction...")
-            board = self.extract_board_from_image("")  # Replace with actual image extraction logic
-            return self.find_best_move(board)
-
+        print(f"Received question: {question[:50]}...")
         if self.maybe_reversed(question):
-            print("Detected likely reversed text. Attempting riddle solving...")
+            print("Detected likely reversed riddle.")
             return self.solve_riddle(question)
+        return "FINAL ANSWER: NOT_A_RIDDLE"
 
-        # Use the model to generate an answer
-        response = self.model.chat_completion(question)  # Assuming this works with raw text input
-        print(f"Agent returning final answer: {response}")
-        return response
-
-
-def run_and_submit_all( profile: gr.OAuthProfile | None):
-    """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
-    and displays the results.
-    """
-    # --- Determine HF Space Runtime URL and Repo URL ---
-    space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
-
-    if profile:
+    def run_and_submit_all( profile: gr.OAuthProfile | None):
+        """
+        Fetches all questions, runs the BasicAgent on them, submits all answers,
+        and displays the results.
+        """
+        # --- Determine HF Space Runtime URL and Repo URL ---
+        space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
+    
+        if profile:
         username= f"{profile.username}"
         print(f"User logged in: {username}")
     else: