import gradio as gr
import os
from openai import AzureOpenAI

client = AzureOpenAI(
  api_key = "0aba6ec588e94eeaa845479f13a49d85",
  api_version = "2024-02-01",
  azure_endpoint = "https://sg-chatkpt.openai.azure.com/"
)


def life_after_death_plan(entity_char, death_type, companion, death_projects):
    response = client.chat.completions.create(
        model="SG_chatDeploy", # model = "gpt-35-turbo".
        messages=[
            {"role": "system", "content": f"Tu t'exprimes sur un ton humoristique et tu te prends pour {entity_char}, spécialiste de la vie dans l'au-delà."},
            {"role": "user", "content": f"Imagine-moi une vie très absurde dans l'au-delà en tenant compte que lors de ma disparition dûe à un(e) {death_type}, j'étais en compagnie de mon ami(e) {companion}, et que ma passion dans la vie était de {death_projects}"}
        ],
        temperature=1
    )
    return response.choices[0].message.content

iface = gr.Interface(
    fn=life_after_death_plan,
    inputs=[
        gr.Radio(["Dieu", "le diable", "Serge Gainsbourg"], label="Vous vous adressez à..."),
        gr.Textbox(lines=1, placeholder="immolation, noyade, chute de pneus...", label="Mort absurde souhaitée"),
        gr.Textbox(lines=1, placeholder="l'être (humain, animal, robot...) qui vous accompagnera dans la mort", label="Votre compagnon de mort"),
        gr.Textbox(lines=2, placeholder="faire du ski avec des chimpanzés", label="Quelle était votre passion dans la vie ?")
    ],
    outputs=gr.Textbox(label="Vos plans dans l'au-delà"),
    title="Quelle sera votre vie après la mort ?",
    description="Ne vous êtes-vous jamais demandé ce qui vous attendait après la mort ? Posez cette question à des gens qui ont les réponses."
)

iface.launch(share=True)