import gradio as gr
from pymongo.mongo_client import MongoClient

ex = [
    ["Quand doit-on choisir une spécialité à l'ISEN ?", "Vous devez choisir une spécialité en fin de 3ème année."], 
    ["Quel est le prix du RU ?", "Le prix du RU est de 3,30€ pour les non-boursiers et de 1€ pour les boursiers."]
]

prompt = """
    "instruction": "{quest}",
    "output": "{resp}"
"""

title = "Veuillez entrer une question et une réponse en rapport avec Junia, l'ISEN, l'école...\nVoir les exemples ci-dessous."
     

def save_to_file(Question, Response):
    try:
        # writing in a file
        with open("data.txt", "a") as file:
            file.write(prompt.format(quest=Question, resp=Response))
        return prompt.format(quest=Question, resp=Response) + "\nEnvoyé avec succès !"

    except Exception as e:
        return "Erreur lors de l'envoi de la question et de la réponse dans la base de données"


site = gr.Interface(fn=save_to_file, inputs=["text", "textbox"], outputs="text", examples=[ex[0], ex[1]], allow_flagging=None, title=title)

site.launch(share=True)