import openai
import gradio as gr

# Configure votre clé API
openai.api_key = "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"

def writing_assistant(debut, suite):
    # Construction de la requête
    prompt = f"DEBUT = '{debut}'\n\n SUITE = '{suite}'\n\n peux-tu réécrire SUITE en t'inspirant du style littéraire de DEBUT, sans préciser d’autres fait que ceux mentionnés dans SUITE ?"

    messages = [
        {"role": "system", "content": f"Tu es un auteur contemporain."},
        {"role": "user", "content": prompt}
    ]

    # Call GPT-3.5-turbo API
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages,
    )

    # Get generated text
    texte_reecrit = response.choices[0].message['content'].strip()

    return texte_reecrit    

# Création de l'interface Gradio
iface = gr.Interface(
    fn=writing_assistant,
    inputs=[
        gr.inputs.Textbox(lines=2, label="Début"),
        gr.inputs.Textbox(lines=5, label="Suite")
    ],
    outputs=gr.outputs.Textbox(label="Texte réécrit"),
    title="Assistant d'écriture",
    description="Réécrit un brouillon en respectant un début avec un style donnés."
)

# Lancer l'interface
iface.launch()