Update app.py

app.py

@@ -1,26 +1,20 @@
-import os
+from transformers import pipeline
 import gradio as gr
-import openai
 
-openai.api_key = os.getenv("OKITAKOY_AI")
-openai.api_base = "https://api.groq.com/openai/v1"
+# Charge un modèle de summarization (résumé)
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-def ask_okitakoy(prompt):
-    try:
-        response = openai.chat.completions.create(
-            model="mixtral-8x7b-32768",
-            messages=[{"role": "user", "content": prompt}]
-        )
-        return response.choices[0].message.content
-    except Exception as e:
-        return f"Erreur : {str(e)}"
+def resumetexte(texte):
+    # Résume le texte (max 130 tokens de résumé)
+    result = summarizer(texte, max_length=130, min_length=30, do_sample=False)
+    return result[0]['summary_text']
 
 iface = gr.Interface(
-    fn=ask_okitakoy,
-    inputs=gr.Textbox(placeholder="Pose ta question à Okitakoy...", lines=2),
+    fn=resumetexte,
+    inputs=gr.Textbox(lines=10, placeholder="Colle ici ton texte à résumer..."),
     outputs="text",
-    title="Okitakoy - Mon Assistant IA (via Groq)",
-    description="Assistant IA rapide, gratuit et intelligent basé sur Mixtral via Groq."
+    title="Résumé automatique",
+    description="Colle un texte, et obtient un résumé généré par BART."
 )
 
 iface.launch()