test

app.py

@@ -1,20 +1,22 @@
 import gradio as gr
-from openai import OpenAI
 import os
+import asyncio
+from anthropic import Anthropic
 from mcp import ClientSession, StdioServerParameters
 from mcp.client.stdio import stdio_client
-import asyncio
 from contextlib import AsyncExitStack
 
-cle_api = os.environ.get("CLE_API_MISTRAL")
-
-# Initialisation du client Mistral (API compatible OpenAI)
-clientLLM = OpenAI(api_key=cle_api, base_url="https://api.mistral.ai/v1")
+# --- Clé API Anthropique ---
+cle_api = os.environ.get("CLE_API_ANTHROPIC")
 
+# Initialisation du client Anthropic
+clientLLM = Anthropic(api_key=cle_api)
 
+# --- Event Loop ---
 loop = asyncio.new_event_loop()
 asyncio.set_event_loop(loop)
 
+# --- Wrapper MCP ---
 class MCPClientWrapper:
     def __init__(self):
         self.session = None
@@ -55,13 +57,15 @@ class MCPClientWrapper:
         tool_names = [tool["name"] for tool in self.tools]
         return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
 
+# --- Connexion au serveur MCP ---
 clientMCP = MCPClientWrapper()
 clientMCP.connect("mcp_server.py")
 print(clientMCP.tools)
 
-# Chatbot : simple écho Fonction chatbot reliée à Mistral
+
+# --- Fonction chatbot reliée à Claude ---
 def chatbot(message, history):
-    # Préparer l’historique dans le format de Mistral
+    # Préparer l’historique pour Anthropique
     messages = []
     for user_msg, bot_msg in history:
         messages.append({"role": "user", "content": user_msg})
@@ -69,17 +73,26 @@ def chatbot(message, history):
     
     messages.append({"role": "user", "content": message})
 
-    # Appel API Mistral
-    response = clientLLM.chat.completions.create(
-        model="mistral-small-latest", 
+    # Appel API Anthropique
+    response = clientLLM.messages.create(
+        model="claude-3-5-sonnet-20240620",   # tu peux changer pour claude-3-opus ou autre
+        max_tokens=500,
         messages=messages,
-        tools=clientMCP.tools
+        tools=clientMCP.tools if clientMCP.tools else None
     )
 
-    bot_reply = response.choices[0].message.content.strip()
+    # Récupération de la réponse
+    bot_reply = ""
+    if response.content:
+        for block in response.content:
+            if block.type == "text":
+                bot_reply += block.text.strip()
+    
     history.append(("Vous: " + message, "Bot: " + bot_reply))
     return history, history
     
+
+# --- UI Gradio ---
 with gr.Blocks() as demo:
 
     chatbot_ui = gr.Chatbot(label="ChatBot")
@@ -87,4 +100,4 @@ with gr.Blocks() as demo:
 
     msg.submit(chatbot, [msg, chatbot_ui], [chatbot_ui, chatbot_ui])
 
-demo.launch(debug=True)
+demo.launch(debug=True)