Update app.py

app.py

@@ -21,8 +21,8 @@ class BasicAgent:
         self.model = genai.GenerativeModel("gemini-2.5-flash")
 
     def __call__(self, question: str) -> str:
-    try:
-        base_prompt = f"""
+        try:
+            base_prompt = f"""
 You are solving benchmark reasoning questions.
 
 RULES:
@@ -36,11 +36,11 @@ Question:
 """
 
         # 🔥 Step 1: First attempt
-        response1 = self.model.generate_content(base_prompt)
-        ans1 = response1.text.strip()
+            response1 = self.model.generate_content(base_prompt)
+            ans1 = response1.text.strip()
 
         # 🔥 Step 2: Self-check (THIS IS THE MAGIC)
-        verify_prompt = f"""
+            verify_prompt = f"""
 Question: {question}
 
 Proposed Answer: {ans1}
@@ -52,18 +52,18 @@ If correct, repeat the answer.
 Return ONLY answer.
 """
 
-        response2 = self.model.generate_content(verify_prompt)
-        final = response2.text.strip()
+            response2 = self.model.generate_content(verify_prompt)
+            final = response2.text.strip()
 
         # 🔥 Clean output (VERY IMPORTANT for exact match)
-        final = final.replace("Final Answer:", "").strip()
-        final = final.split("\n")[0].strip()
-        final = final.split(".")[0].strip()
+            final = final.replace("Final Answer:", "").strip()
+            final = final.split("\n")[0].strip()
+            final = final.split(".")[0].strip()
 
-        return final[:100]
+            return final[:100]
 
-    except Exception as e:
-        return "N/A"
+        except Exception as e:
+            return "N/A"