NicolasGaudemet commited on
Commit
07471eb
1 Parent(s): a3c6d7d

add application file

Browse files
Files changed (1) hide show
  1. writing_assistant_app.py +40 -0
writing_assistant_app.py ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ import gradio as gr
3
+
4
+ # Configure votre clé API
5
+ openai.api_key = "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"
6
+
7
+ def writing_assistant(debut, suite):
8
+ # Construction de la requête
9
+ 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 ?"
10
+
11
+ messages = [
12
+ {"role": "system", "content": f"Tu es un auteur contemporain."},
13
+ {"role": "user", "content": prompt}
14
+ ]
15
+
16
+ # Call GPT-3.5-turbo API
17
+ response = openai.ChatCompletion.create(
18
+ model="gpt-3.5-turbo",
19
+ messages=messages,
20
+ )
21
+
22
+ # Get generated text
23
+ texte_reecrit = response.choices[0].message['content'].strip()
24
+
25
+ return texte_reecrit
26
+
27
+ # Création de l'interface Gradio
28
+ iface = gr.Interface(
29
+ fn=writing_assistant,
30
+ inputs=[
31
+ gr.inputs.Textbox(lines=2, label="Début"),
32
+ gr.inputs.Textbox(lines=5, label="Suite")
33
+ ],
34
+ outputs=gr.outputs.Textbox(label="Texte réécrit"),
35
+ title="Assistant d'écriture",
36
+ description="Réécrit un brouillon en respectant un début avec un style donnés."
37
+ )
38
+
39
+ # Lancer l'interface
40
+ iface.launch()