import gradio as gr
import os
import pandas as pd

from typing import Dict, List, Optional
from model import generate_response
from datasets import load_dataset
from huggingface_hub import login
from huggingface_hub import whoami
from huggingface_hub import dataset_info


DATASET_CONFIGS = [
    'covidqa', 'cuad', 'delucionqa', 'emanual', 'expertqa', 'finqa', 'hagrid', 'hotpotqa', 'msmarco', 'pubmedqa','tatqa', 'techqa'
]

hf_token = os.getenv("HF_TOKEN")
login(hf_token)

def chatbot(prompt):
    return whoami()

iface = gr.Interface(fn=chatbot, inputs="text", outputs="text", title="Capstone Project Group 10")

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