import os
import gradio as gr
import requests
from TwitterChatBot.main import ask


def get_answer(question):
    answer = ask(question=question)
    return answer.strip()


# def get_answer(question):
#     try:
#         answer = requests.get(
#             url,
#             json={"question": question},
#         )
#     except Exception as err:
#         return f"Sorry there was a problem with {err}, please check your connection and try again."

#     if answer.status_code == 200:
#         return answer.json()["answer"]

#     return "Sorry, We have a problem with our server"


def predict(input, history=[]):
    answer = get_answer(input)
    history.append((input, answer))
    response = history
    return response, history


with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    state = gr.State([])

    with gr.Row():
        txt = gr.Textbox(
            show_label=False, placeholder="Enter text and press enter"
        ).style(container=False)

    txt.submit(predict, [txt, state], [chatbot, state])

demo.launch()