writing_assistant_app.py

import openai
import os 
import gradio as gr

# Configure votre clé API
openai.api_key = os.environ['OpenaiKey']

def writing_assistant(debut, suite, instructions):
    # Construction de la requête
    prompt = f"DEBUT = '{debut}'\n\n SUITE = '{suite}'\n\n
            Peux-tu réécrire SUITE en reprenant le style littéraire de DEBUT
            (ton, rythme, registre de vocabulaire, type de mots, taille des phrases, ponctuation...) ?
            N'ajoute pas d'autres éléments narratifs que ceux mentionnés dans SUITE.
            \n\n {instructions}"

    messages = [
        {"role": "system", "content": f"Tu es un assistant d'écriture. Tu aides un auteur contemporain à écrire, en t'inspirant de son style littéraire."},
        {"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=5, label="Début"),
        gr.inputs.Textbox(lines=5, label="Suite")
        gr.inputs.Textbox(lines=2, label="Instructions additionnelles")
    ],
    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é."
)

# Lancer l'interface
iface.launch()