fin_assist / app.py
skumar9's picture
Update app.py
d227bc9
raw
history blame
1.88 kB
import logging
logging.basicConfig()
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
logger.info("Loaded " + __name__)
import gradio as gr
import random
import time
import requests
import json
import uuid
logger.info(100*"#")
logger.info("Initializing gradio!")
get_window_url_params = """
function(text_input, url_params) {
console.log(text_input, url_params);
const params = new URLSearchParams(window.location.search);
url_params = Object.fromEntries(params);
return [text_input, url_params];
}
"""
with gr.Blocks(theme=gr.themes.Soft()) as demo:
import os
import requests
import json
import uuid
USER=100
SESS_ID=str(uuid.uuid1())
url="http://ds-lang-server:8777/v1/user_conv"
CONV_API=os.environ.get("CONV_API",url)
def call_api(user_input,user_id,session_id):
url = CONV_API
payload = json.dumps({
"user_id": USER,
"session_id":SESS_ID,
"user_input": user_input
})
headers = {
'accept': 'application/json',
'Content-Type': 'application/json',
'Authorization':os.environ.get('AUTH_TOKEN', None)
}
response = requests.request("POST", url, headers=headers, data=payload)
ret=response.json()
return ret['result']
def respond(message, chat_history,url_params):
print("params",url_params)
bot_message=call_api(message,12,"xyz")
chat_history.append((message, bot_message))
return "", chat_history
chatbot = gr.Chatbot(label='Jivi')
msg = gr.Textbox(label='Patient')
clear = gr.ClearButton([msg, chatbot])
msg.submit(respond, [msg, chatbot,url_params], [msg, chatbot],_js=get_window_url_params)
demo.launch(auth=(os.environ.get("USER","demo"), os.environ.get("PASSWORD","demo")))