Update app.py

app.py

@@ -11,10 +11,8 @@ from smolagents import CodeAgent, HfApiModel, tool
 def get_current_time_in_timezone(timezone: Optional[str] = None) -> str:
     """
     Returns the current time. If a timezone is provided, it is noted in the message.
-
     Args:
         timezone: An optional timezone name or location.
-
     Returns:
         A string with the current time.
     """
@@ -28,10 +26,8 @@ def get_current_time_in_timezone(timezone: Optional[str] = None) -> str:
 def parse_csv(file_path: str) -> str:
     """
     Parses a CSV file and returns the first few rows.
-
     Args:
         file_path: Path to the CSV file to be parsed.
-
     Returns:
         A preview of the first rows.
     """
@@ -43,10 +39,8 @@ def parse_csv(file_path: str) -> str:
 def transcribe_audio(audio_path: str) -> str:
     """
     Transcribes spoken words in an audio file.
-
     Args:
         audio_path: Path to the .mp3 or .wav file.
-
     Returns:
         A transcription of the audio.
     """
@@ -73,7 +67,12 @@ class BasicAgent:
         print("✅ Agent initialized with all tools")
 
     def __call__(self, question: str) -> str:
-        return self.agent.run(question)
+        response = self.agent.run(question)
+        if "Final Answer:" not in response:
+            # Add fallback final answer if missing
+            last_line = response.strip().split("\n")[-1]
+            response += f"\nFinal Answer: {last_line}"
+        return response
 
 # === Evaluation Logic ===
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -147,6 +146,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             f"Overall Score: {result_data.get('score', 'N/A')}% "
             f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
             f"Message: {result_data.get('message', 'No message received.')}"
+
         )
         return final_status, pd.DataFrame(results_log)
     except Exception as e: