File size: 1,226 Bytes
07471eb 6254bf5 07471eb 6254bf5 07471eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
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=5, 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é."
)
# Lancer l'interface
iface.launch() |