|
|
from __future__ import annotations |
|
|
|
|
|
import sys |
|
|
from io import StringIO |
|
|
from typing import Annotated |
|
|
|
|
|
import gradio as gr |
|
|
|
|
|
from app import _log_call_end, _log_call_start, _truncate_for_log |
|
|
|
|
|
|
|
|
def Code_Interpreter(code: Annotated[str, "Python source code to run; stdout is captured and returned."]) -> str: |
|
|
_log_call_start("Code_Interpreter", code=_truncate_for_log(code or "", 300)) |
|
|
if code is None: |
|
|
result = "No code provided." |
|
|
_log_call_end("Code_Interpreter", result) |
|
|
return result |
|
|
old_stdout = sys.stdout |
|
|
redirected_output = sys.stdout = StringIO() |
|
|
try: |
|
|
exec(code) |
|
|
result = redirected_output.getvalue() |
|
|
except Exception as exc: |
|
|
result = str(exc) |
|
|
finally: |
|
|
sys.stdout = old_stdout |
|
|
_log_call_end("Code_Interpreter", _truncate_for_log(result)) |
|
|
return result |
|
|
|
|
|
|
|
|
def build_interface() -> gr.Interface: |
|
|
return gr.Interface( |
|
|
fn=Code_Interpreter, |
|
|
inputs=gr.Code(label="Python Code", language="python"), |
|
|
outputs=gr.Textbox(label="Output", lines=5, max_lines=20), |
|
|
title="Code Interpreter", |
|
|
description="<div style=\"text-align:center\">Execute Python code and see the output.</div>", |
|
|
api_description=( |
|
|
"Execute arbitrary Python code and return captured stdout or an error message. " |
|
|
"Supports any valid Python code including imports, variables, functions, loops, and calculations. " |
|
|
"Examples: 'print(2+2)', 'import math; print(math.sqrt(16))', 'for i in range(3): print(i)'. " |
|
|
"Parameters: code (str - Python source code to execute). " |
|
|
"Returns: Combined stdout output or exception text if execution fails." |
|
|
), |
|
|
flagging_mode="never", |
|
|
) |
|
|
|
|
|
|
|
|
__all__ = ["Code_Interpreter", "build_interface"] |
|
|
|
