from langchain.agents.openai_assistant import OpenAIAssistantRunnable
import os

asst_id = os.getenv('assistant_key')
#key = os.getenv('open_ai')

interpreter_assistant = OpenAIAssistantRunnable(assistant_id=asst_id)
import gradio as gr

def chat_response(message, history):
  output = interpreter_assistant.invoke({"content": message})
  response = output[0].content[0].text.value
  return response
css = """
label[data-testid="block-label"] {
    display: none !important;
}
footer {
    display: none !important;
}
"""

js_func = """
function refresh() {
    const url = new URL(window.location);
    if (url.searchParams.get('__theme') !== 'dark') {
        url.searchParams.set('__theme', 'dark');
        window.location.href = url.href;
    }
}
"""
with gr.Blocks(css=css, js = js_func, theme="monochrome") as demo:
  chatbot = gr.ChatInterface(chat_response,title="ProjSite")

demo.launch(share=True)