import gradio as gr
import requests

api_url = 'https://huynhdoo--sirius-api-prompt.modal.run'

origins = ['peurChangementConseil',
 'trouverEntrepriseConseil',
 'choseMarquanteConseil',
 'differenceCollegeCfaConseil',
 'descriptionMetierConseil',
 'commentTrouverEntrepriseAutre',
 'avantFormationAutre',
 'difficileEntrepriseAutre']

def response(origin, topic):

    json = dict(
        file_id='1yLHGKfKWlOKklqNKsVQb41eRyCbCzmwY',
        topic=topic,
        origin=origin
    )

    resp = requests.post(url=api_url, json=json)
    data = resp.json()
    return data['response']

gradio_app = gr.Interface(
    fn=response,
    inputs=[
        gr.Dropdown(origins, label="Origine", info="Choisir un type de conseil"),
        gr.Textbox(label="Génération", info="Décrire le résultat attendu")
    ],
    outputs=[
        gr.Textbox(label="Réponse")
    ],
    examples=[['descriptionMetierConseil', 'liste en français sans répétition des 10 principaux verbes d\'actions'],
              ['trouverEntrepriseConseil', 'liste en français sans répétition des 10 qualités principales détaillées'],
              ['choseMarquanteConseil', 'résumé synthétique en français des qualités requises en 3 paragraphes séparés'],
              ['differenceCollegeCfaConseil', 'résumé synthétique en français des points négatifs en 3 paragraphes séparés'],
              ['peurChangementConseil', 'résumé synthétique en français des conseils en 3 paragraphes séparés'],
              ['descriptionMetierConseil', 'résumé en français des principaux métiers en 3 paragraphes séparés'],
    ]
)

if __name__ == "__main__":
    gradio_app.launch(auth=("sirius", "black"), cache_examples=False)