from config import json_url_id
from main import run_main
import gdown
import gradio as gr

download_url = f'https://drive.google.com/uc?id={json_url_id}'
output = 'secret_google_service_account.json'
gdown.download(download_url, output, quiet=False)

with gr.Blocks(title="Accountant automation",theme='nota-ai/theme',css="footer {visibility: hidden}") as demo:
            
    with gr.Row():
        with gr.Column(scale=6):
            with gr.Row():
                with gr.Column(scale=3):
                    gr.Markdown("Pour pouvoir connecter votre Google Drive a cette application, veuillez partager les dossiers avec l'addresse email suivante:\n`synthaialphav2@gen-lang-client-0635767828.iam.gserviceaccount.com`\n\nPour le fichier vers le relevé de compte, veillez à ce que le champ `Accès géneral` soit bien paramétré à\n`Tous les utilisateurs qui ont le lien`")
                    source_folder_with_reciepts = gr.Textbox(label="Dossier contenant les factures", lines=1)
                    link_to_csv = gr.Textbox(label="Lien vers le relevé de compte en csv", lines=1, )
                    folder_to_save_processed_reciepts = gr.Textbox(label="Dossier où sauvegarder les factures", lines=1, )
                    folder_to_save_reconciled_data = gr.Textbox(label="Dossier où sauvegarder le tableau final", lines=1, )
                    name_output_file = gr.Textbox(label="Nom du fichier de tableau final", lines=1, )
                    transaction_csv_path = gr.Textbox(label="Ne pas modifier ce champ", value="downloaded_file.csv", lines=1,visible=True )
                    chat_submit_button = gr.Button(value="Valider ▶")
                    
                with gr.Column(scale=6):
                    chat_output = gr.Markdown("Appuyez sur valider pour lancer le processing")

                    
    fn_chat = run_main

    
    chat_submit_button.click(fn=fn_chat, inputs=[source_folder_with_reciepts, link_to_csv, folder_to_save_processed_reciepts, folder_to_save_reconciled_data, name_output_file, transaction_csv_path], outputs=[chat_output])

demo.launch(max_threads=40)