créationde chatbot

Dockerfile

@@ -1,18 +1,11 @@
-FROM python:3.11-slim
-
-# Installer les dépendances
-RUN pip install litellm fastapi uvicorn pydantic
+FROM python:3.9
 
 WORKDIR /app
 
-# Copier les fichiers nécessaires
-COPY app.py .
+COPY requirements.txt .
 
-# Variables d'environnement
-ENV GEMINI_API_KEY=${GEMINI_API_KEY}
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Exposer le port
-EXPOSE 7860
+COPY . .
 
-# Commande de démarrage
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]

app.py

@@ -2,6 +2,7 @@ import os
 import litellm
 from fastapi import FastAPI, HTTPException, Header, Depends
 from pydantic import BaseModel
+import gradio as gr
 
 # Configuration du modèle
 model_name = "gemini/gemini-1.5-pro"
@@ -28,6 +29,30 @@ async def verify_api_key(x_api_key: str = Header(...)):
     if x_api_key != USER_API_KEY:
         raise HTTPException(status_code=401, detail="Clé API invalide")
 
+# Historique de la conversation
+conversation_history = []
+
+def chatbot_function(user_input):
+    conversation_history.append({"role": "user", "content": user_input})
+
+    response = litellm.completion(
+        model=model_name,  # Utilisation du modèle par défaut pour le chatbot
+        messages=conversation_history
+    )
+
+    conversation_history.append({"role": "assistant", "content": response.choices[0].message.content})
+
+    return conversation_history
+
+# Interface Gradio
+iface = gr.Interface(
+    fn=chatbot_function,
+    inputs=gr.Textbox(lines=2, placeholder="Votre message..."),
+    outputs="chatbot",
+    title="Chatbot",
+)
+
+
 @app.post("/", dependencies=[Depends(verify_api_key)])
 async def predict(request: PromptRequest):
     try:
@@ -41,3 +66,9 @@ async def predict(request: PromptRequest):
         return {"response": response.choices[0].message.content, "model": request.model}
     except Exception as e:
         return {"error": str(e)}
+
+# Lancement de l'interface Gradio après le lancement de FastAPI
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)  # Lance FastAPI
+    iface.launch(share=True)  # Lance Gradio et partage l'interface

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+pydantic
+litellm
+python-multipart
+gradio