Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+
 from langgraph.graph import StateGraph, END
 from typing import TypedDict
 import string
@@ -21,6 +22,7 @@ class AgentState(TypedDict):
     response: str
     is_reversed: bool
     is_riddle: bool
+    is_python: bool
 
 class SuperSmartAgent:
     def __init__(self):
@@ -33,6 +35,8 @@ class SuperSmartAgent:
         workflow.add_node("fix_question", self.fix_question)
         workflow.add_node("check_riddle_or_trick", self.check_riddle_or_trick)
         workflow.add_node("solve_riddle", self.solve_riddle)
+        workflow.add_node("check_python_suitability", self.check_python_suitability)
+        workflow.add_node("execute_python_code", self.execute_python_code)
 
         workflow.set_entry_point("check_reversed")
 
@@ -42,10 +46,15 @@ class SuperSmartAgent:
         )
         workflow.add_conditional_edges(
             "check_riddle_or_trick",
-            lambda state: "solve_riddle" if state["is_riddle"] else END,
+            lambda state: "solve_riddle" if state["is_riddle"] else "check_python_suitability",
+        )
+        workflow.add_conditional_edges(
+            "check_python_suitability",
+            lambda state: "execute_python_code" if state["is_python"] else END,
         )
         workflow.add_edge("fix_question", "check_riddle_or_trick")
         workflow.add_edge("solve_riddle", END)
+        workflow.add_edge("execute_python_code", END)
 
         return workflow.compile()
 
@@ -54,7 +63,8 @@ class SuperSmartAgent:
             question=question,
             response="",
             is_reversed=False,
-            is_riddle=False
+            is_riddle=False,
+            is_python=False
         )
         final_state = self.graph.invoke(initial_state)
         return final_state["response"]
@@ -106,6 +116,39 @@ class SuperSmartAgent:
             state["response"] = "Could not solve riddle."
         return state
 
+    def check_python_suitability(self, state):
+        question = state["question"].lower()
+        patterns = ["output", "python", "execute", "run", "code", "script"]
+        state["is_python"] = any(word in question for word in patterns)
+        return state
+
+    def execute_python_code(self, state):
+        question = state["question"]
+
+        # Check if the question is about executing Python code
+        if "output from the attached Python code" in question.lower():
+            # Placeholder for actual file handling and execution logic
+            # In a real implementation, you would read the attached file and execute it
+            python_code = """
+                # Example Python code
+                def calculate():
+                    return 42
+                
+                result = calculate()
+                print(result)
+                """
+            try:
+                # Use code_interpreter to execute the Python code
+                result = code_interpreter(code=python_code)
+                state["response"] = f"The final numeric output is: {result}"
+            except Exception as e:
+                state["response"] = f"Error executing Python code: {str(e)}"
+        else:
+            state["response"] = "This question does not seem to involve executing Python code."
+
+        return state
+
+