unhcr / app.py
elyxlz
initial commit
0c7add2
raw
history blame
2.63 kB
import os
from typing import Optional, Tuple
import gradio as gr
import argparse
import datetime
import pickle
#import whisper
import dotenv
import sys
from io import StringIO
import re
dotenv.load_dotenv()
from langchain.callbacks import get_openai_callback
import hydra
from omegaconf import DictConfig, open_dict, OmegaConf
class ChatbotAgentGradio():
def __init__(
self,
config_name
):
config = OmegaConf.load(f'./config/{config_name}.yaml')
self.chatbot = hydra.utils.instantiate(config.model, _convert_="partial")
def chat(self,
inp: str,
history: Optional[Tuple[str, str]],
):
"""Method for integration with gradio Chatbot"""
print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
print("inp: " + inp)
history = history or []
output = self.chatbot.run(inp)
history.append((inp, output))
return history, history#, ""
def update_foo(self, widget, state):
if widget:
state = widget
return state
def launch_app(self):
block = gr.Blocks(css=".gradio-container {background-color: lightgray}")
with block:
instance = gr.State()
show_chain_state = gr.State(False)
with gr.Row():
gr.Markdown("<h3><center>UNHCR</center></h3>")
with gr.Row():
chatbot = gr.Chatbot()
with gr.Row():
message = gr.Textbox(
label="What's your question?",
lines=1,
)
submit = gr.Button(value="Send", variant="secondary").style(full_width=False)
state = gr.State()
agent_state = gr.State()
submit.click(self.chat, inputs=[message, state], outputs=[chatbot, state])
message.submit(self.chat, inputs=[message, state], outputs=[chatbot, state])
block.launch(debug=True, share=False, server_port=7861)#, server_name='192.168.0.73', )
def simple(config):
config = OmegaConf.load(f'./config/{config}.yaml')
chatbot = hydra.utils.instantiate(config.model, _convert_="partial")
while True:
inp = input("\nUser: ")
print(chatbot.run(inp))
if __name__ == '__main__':
#simple('conf_0.1')
app = ChatbotAgentGradio('conf_0.1')
app.launch_app()
#QA = QA(store, k=1)
#app = QAGradio(QA)
#app.launch_app()