Update app.py

app.py

@@ -257,13 +257,55 @@ class GaiaAgent:
                 file_path = download_file(file_url, self.temp_dir, task_id)
                 if file_path:
                     try:
-                        with open(file_path, 'r') as f: python_code = f.read()
-                        python_tool = PythonREPLTool()
-                        exec_output = python_tool.run(python_code)
-                        response = self.llm.invoke([HumanMessage(content=f"Python output: ```{exec_output}``` What is final numeric output? Only the number.")])
-                        analysis_result = response.content.strip()
-                    except Exception as e: analysis_result = f"ERROR: Python execution failed. {e}"
-                else: analysis_result = "ERROR: Failed download."
+                        # Use subprocess to run the script and capture output reliably
+                        logging.info(f"Executing Python script using subprocess: {file_path}")
+                        # Ensure using the correct python executable for the environment
+                        import sys
+                        process = subprocess.run(
+                            [sys.executable, str(file_path)], # Use python executable from sys
+                            capture_output=True, # Capture stdout and stderr
+                            text=True,           # Decode stdout/stderr as text
+                            timeout=15,          # Add a reasonable timeout
+                            check=False          # Don't raise exception on non-zero exit code
+                        )
+
+                        stdout = process.stdout.strip()
+                        stderr = process.stderr.strip()
+
+                        if process.returncode != 0:
+                            # Script failed
+                            logging.error(f"Python script {file_path} failed (Code: {process.returncode}). Stderr: {stderr}")
+                            analysis_result = f"ERROR: Python script failed with code {process.returncode}. Error: {stderr}"
+                        elif not stdout and stderr:
+                            # Script ran but only produced error messages
+                             logging.warning(f"Python script {file_path} succeeded but produced only stderr: {stderr}")
+                             analysis_result = f"ERROR: Python script produced errors: {stderr}"
+                        elif not stdout:
+                             # Script ran but produced no output at all
+                             logging.warning(f"Python script {file_path} produced no standard output.")
+                             analysis_result = "ERROR: Python script produced no output."
+                        else:
+                            # Script succeeded and produced output, assume stdout is the answer
+                            logging.info(f"Python script {file_path} executed. Output: {stdout}")
+                            analysis_result = stdout
+                            # Optional: Validate if it looks like a number, but exact match might require raw output
+                            try:
+                                float(analysis_result) # Simple check
+                            except ValueError:
+                                logging.warning(f"Python script output '{analysis_result}' may not be purely numeric.")
+                                # Still return the raw output as it might be the expected format
+
+                    except FileNotFoundError:
+                         logging.error(f"Python executable '{sys.executable}' not found? Error running script.")
+                         analysis_result = "ERROR: Python interpreter not found."
+                    except subprocess.TimeoutExpired:
+                         logging.error(f"Python script {file_path} timed out after 15 seconds.")
+                         analysis_result = "ERROR: Python script execution timed out."
+                    except Exception as e:
+                         logging.error(f"Error executing Python script {file_path} via subprocess: {e}", exc_info=True)
+                         analysis_result = f"ERROR: Failed to execute Python script. Details: {str(e)}"
+                else:
+                    analysis_result = "ERROR: Failed to download Python code file."
             # Q14: Calculus Audio
             elif "homework.mp3" in q_lower:
                 file_path = download_file(file_url, self.temp_dir, task_id)