Update writing_assistant_app.py

writing_assistant_app.py

@@ -5,12 +5,16 @@ import gradio as gr
 # Configure votre clé API
 openai.api_key = os.environ['OpenaiKey']
 
-def writing_assistant(debut, suite):
+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 t'inspirant du style littéraire de DEBUT, sans préciser d’autres fait que ceux mentionnés dans SUITE ?"
+    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 auteur contemporain."},
+        {"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}
     ]
 
@@ -31,6 +35,7 @@ iface = gr.Interface(
     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",