from transformers import pipeline, Conversation
import gradio as gr

chatbot = pipeline(model="Falconsai/medical_summarization")

message_list = []
response_list = []

# def vanilla_chatbot(message, history):
#     conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
#     conversation = chatbot(conversation)

#     return conversation.generated_responses[-1]

# demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Test medical summarization chatbot", description="Enter text to summarize.")

def responseFun(message, history):
    return history[len(history) - 1]
    
demo_chatbot = gr.ChatInterface(responseFun, title="Test medical summarization chatbot", description="Enter text to summarize.")

demo_chatbot.launch()