ysharma's picture
ysharma HF Staff
Update app.py
7312f21
raw
history blame
1.12 kB
import interpreter
import gradio as gr
import os
import io
import contextlib
# Paste your OpenAI API key below.
interpreter.api_key = os.environ.get('openai_api_key')
interpreter.auto_run = True
def chat_with_interpreter(message, history=[]):
if message == 'reset':
interpreter.reset()
# Redirect stdout to capture the streamed output
new_stdout = io.StringIO()
with contextlib.redirect_stdout(new_stdout):
interpreter.chat(message)
output = new_stdout.getvalue()
# Return this output so Gradio's ChatInterface can display it
return output
demo = gr.ChatInterface(fn=chat_with_interpreter,
title="Open-Interpreter Gradio ChatInterface",
description="Open Interpreter lets LLMs run code (Python, Javascript, Shell, and more) locally",
examples=["what is 2+2?", "Can you solve for x: 10x -65=0?", "What are top 10 headlines from BBC from last week?"],
clear_btn=None,
retry_btn=None,
undo_btn=None).queue()
demo.launch(debug=True)