Update chatbot.py

chatbot.py

@@ -1,70 +1,58 @@
-import streamlit as st
+# chatbot.py
+
 import cohere
 import requests
 from io import BytesIO
-import base64
-
-
-# Configuration de la clé API
-COHERE_API_KEY = "moZJbgxiW9cW8Wqo0ecce0pa84uf3eT6F2oL1whB"  # Remplace par ta clé API
-co = cohere.Client(COHERE_API_KEY)
-
-# Initialisation du contexte de chat
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-
-st.set_page_config(page_title="Chatbot Cohere", page_icon="💬")
-st.title("💬 Chat avec Cohere (Command R+)")
-
-# Boîte d'entrée utilisateur
-user_input = st.chat_input("Pose ta question...")
-
-# Affichage de l'historique visuel
-for msg in st.session_state.chat_history:
-    with st.chat_message("user" if msg["role"] == "USER" else "assistant"):
-        st.markdown(msg["text"])
-
-# Quand l'utilisateur envoie un message
-if user_input:
-    with st.chat_message("user"):
-        st.markdown(user_input)
-
-    st.session_state.chat_history.append({"role": "USER", "text": user_input})
-
-    # Création de l'historique pour Cohere
-    cohere_history = [
-        {"role": msg["role"], "message": msg["text"]}
-        for msg in st.session_state.chat_history
-        if msg["role"] in ["USER", "CHATBOT"]
-    ]
 
-    try:
-        response = co.chat(
-            message=user_input,
-            chat_history=cohere_history,
-            model="command-r-plus",
-            temperature=0.7,
-            max_tokens=300
-        )
-
-        reply = response.text.strip()
-
-        # Étape 1 : Résumer la réponse avec Cohere
-        summary_prompt = f"Résume en une seule phrase simple et claire en français ce texte : {reply}"
-        summary_response = co.generate(prompt=summary_prompt, max_tokens=100)
-        summary = summary_response.generations[0].text.strip()
-
-        # Étape 2 : Appel ElevenLabs (TTS)
-        ELEVEN_API_KEY = "sk_6c5472c80964f88fcdc9d9c189db749143ae1a0d2c8f26f3"  # ← Remplace par ta clé API ElevenLabs
-        voice_id = "TxGEqnHWrfWFTfGW9XjX"  # Voix française (Antoine)
+class InflationChatbot:
+    def __init__(self, cohere_api_key, eleven_api_key):
+        self.co = cohere.Client(cohere_api_key)
+        self.eleven_api_key = eleven_api_key
+        self.voice_id = "TxGEqnHWrfWFTfGW9XjX"  # Antoine (voix française)
+
+    def ask(self, user_input, chat_history=[]):
+        try:
+            # Reformatage de l’historique
+            cohere_history = [
+                {"role": msg["role"], "message": msg["text"]}
+                for msg in chat_history
+                if msg["role"] in ["USER", "CHATBOT"]
+            ]
+
+            # Réponse du modèle Cohere
+            response = self.co.chat(
+                message=user_input,
+                chat_history=cohere_history,
+                model="command-r-plus",
+                temperature=0.7,
+                max_tokens=300
+            )
+            reply = response.text.strip()
+
+            # Résumé
+            summary_prompt = f"Résume en une seule phrase simple et claire en français ce texte : {reply}"
+            summary_response = self.co.generate(prompt=summary_prompt, max_tokens=100)
+            summary = summary_response.generations[0].text.strip()
+
+            # Synthèse vocale via ElevenLabs
+            audio = self.text_to_speech(summary)
+
+            return {
+                "reply": reply,
+                "summary": summary,
+                "audio": audio
+            }
+        except Exception as e:
+            return {"error": str(e)}
 
+    def text_to_speech(self, text):
         headers = {
-            "xi-api-key": ELEVEN_API_KEY,
+            "xi-api-key": self.eleven_api_key,
             "Content-Type": "application/json"
         }
 
         data = {
-            "text": summary,
+            "text": text,
             "model_id": "eleven_multilingual_v2",
             "voice_settings": {
                 "stability": 0.5,
@@ -72,25 +60,10 @@ if user_input:
             }
         }
 
-        response_audio = requests.post(
-            f"https://api.elevenlabs.io/v1/text-to-speech/{voice_id}",
+        response = requests.post(
+            f"https://api.elevenlabs.io/v1/text-to-speech/{self.voice_id}",
             headers=headers,
             json=data
         )
 
-        # Affichage de la réponse texte + audio
-        with st.chat_message("assistant"):
-            st.markdown(reply)
-            st.markdown(f"🔊 **Résumé vocal :** _{summary}_")
-
-            if response_audio.status_code == 200:
-                audio_bytes = BytesIO(response_audio.content)
-                st.audio(audio_bytes, format="audio/mp3")
-            else:
-                st.warning("Erreur dans la synthèse vocale (ElevenLabs).")
-
-        st.session_state.chat_history.append({"role": "CHATBOT", "text": reply})
-
-    except Exception as e:
-        st.error(f"Erreur : {str(e)}")
-
+        return BytesIO(response.content) if response.status_code == 200 else None