Update app.py

app.py

@@ -19,42 +19,33 @@ chatbot = pipeline(
     device_map="auto"
 )
 
-system_prompt = """Tu es Aria, une IA bienveillante, claire et polie.
-Réponds directement aux questions en une ou plusieurs phrases complètes.
-Ne répète jamais le message de l'utilisateur. Ne joue pas un rôle.
-N'inclus pas "Utilisateur:" ou "Aria:" dans ta réponse."""
+system_prompt = """Tu es Aria, une IA gentille et claire. 
+Réponds directement à la question de l'utilisateur en français, 
+en une ou deux phrases complètes. 
+Ne répète pas la question. Ne joue pas de rôle."""
 
 def clean_reply(text):
-    # Supprimer tout ce qui pourrait être un rôle ou répétition
-    text = re.sub(r"^.*?Aria:\s*", "", text, flags=re.DOTALL)
-    text = re.sub(r"Utilisateur:.*", "", text, flags=re.DOTALL)
-    text = text.strip()
-    return text
+    # Retirer tout texte avant ou après la vraie réponse
+    text = re.sub(r"^.*?:", "", text, flags=re.DOTALL)  # supprime jusqu'au premier ":"
+    return text.strip()
 
-def chat(message, history=[]):
-    history = history[-3:]
-
-    # Contexte plus naturel
-    context = "\n".join([f"{m[0]}\nRéponse: {m[1]}" for m in history])
-    prompt = f"{system_prompt}\n\n{context}\n{message}\nRéponse:"
+def chat(message):
+    prompt = f"{system_prompt}\n\nQuestion : {message}\nRéponse :"
 
     resp = chatbot(
         prompt,
-        max_new_tokens=120,
+        max_new_tokens=80,
         do_sample=True,
         temperature=0.7,
-        top_p=0.9,
-        repetition_penalty=1.1
+        top_p=0.9
     )[0]["generated_text"]
 
     reply = clean_reply(resp)
-    history.append([message, reply])
-
-    return history, history
+    return reply
 
 with gr.Blocks() as demo:
-    chat_ui = gr.Chatbot()
     msg = gr.Textbox(placeholder="Écris un message...")
-    msg.submit(chat, [msg, chat_ui], [chat_ui, chat_ui])
+    output = gr.Textbox(label="Réponse d'Aria")
+    msg.submit(lambda m: chat(m), msg, output)
 
 demo.launch()