# App.py to launch the app via hugging face
#######################################################################################################
# IMPORT
#######################################################################################################

import pandas as pd
import gradio as gr

# modules
from modules.language_model import TAPAS


##################################################################################
# Function that enables testing
##################################################################################
table_main = pd.read_csv('./data' + '/' + 'gadm41_DEU_1_main').astype(str)

def AskAI(ques, lv, table_main = table_main):
    level = int(lv) # Currently placeholder
    question = str(ques)
    ans = TAPAS(question = question, table_main= table_main)
    return(ans)

def AskAI_easy(ques):
    Tmain = pd.read_csv('./data' + '/' + 'gadm41_DEU_1_main').astype(str)
    blub = str(AskAI(ques,1,Tmain))
    return(blub)

    
#######################################################################################
# Gradio Interface
###############################################################################
desc = 'Example: What is the Bundesland with tyhe biggest Area?'
iface = gr.Interface(fn=AskAI_easy, inputs=['text'], outputs='text', description= desc)
iface.launch()