import gradio as gr
import csv
from transformers import pipeline

responses = {}

with open('Answers.csv', mode='r', encoding='utf-8-sig') as file:
    reader = csv.DictReader(file)
    for row in reader:
        intent = row['Intent']
        answer = row['Answer']
        
        responses[intent] = answer

def respond(message, history):
    pipe = pipeline("text-classification", model="sehrishsafdar/urdu-intent-model")
    intent = pipe.predict(message)
    print(intent)
    return responses[intent[0]['label']]

demo = gr.ChatInterface(
    respond, 
    title="Urdu University Information Desk Chatbot",
)


if __name__ == "__main__":
    demo.launch()