add code_runner

app.py

@@ -89,6 +89,35 @@ import torch
 from codeinterpreterapi import CodeInterpreterSession
 import html2text
 
+from interpreter.code_interpreter import CodeInterpreter
+from interpreter.code_block import CodeBlock
+
+class CodeBlock:
+    def __init__(self, code):
+        self.code = code
+        self.output = ""
+        self.active_line = None
+
+    def refresh(self):
+        print(f"Active line: {self.active_line}")
+        print(f"Output: {self.output}")
+
+
+code_raw = """
+for i in range(3):
+  print("hello world")
+"""
+def Code_Runner(code_raw: str, lang: str='python'):
+    # interpreter = CodeInterpreter(language="python", debug_mode=True)
+    interpreter = CodeInterpreter(language=lang, debug_mode=True)
+    code_block = CodeBlock(code_raw)
+    interpreter.active_block = code_block
+    output = interpreter.run()
+    print("Real Output: \n", output)
+    return output
+
+Code_Runner(code_raw)
+
 async def TestCodeInterpret(CustomMessage:str):
     # create a session
     session = CodeInterpreterSession(llm=GPTfake)

requirements.txt

@@ -25,3 +25,4 @@ scipy
 codeinterpreterapi
 jupyter-kernel-gateway
 html2text
+open-interpreter