Update app.py

app.py

@@ -1,13 +1,12 @@
 from fastapi import FastAPI, UploadFile, File
 import requests
-import os
 
 app = FastAPI()
 
-# URL de l’API Ollama (par défaut, elle tourne localement sur le port 11434)
+# URL de l’API Ollama (local dans le conteneur)
 OLLAMA_API_URL = "http://localhost:11434/api/generate"
 
-# Fonction pour appeler Ollama avec un prompt
+# Fonction pour interroger Ollama
 def query_ollama(prompt: str, model: str = "mistral"):
     payload = {
         "model": model,
@@ -22,26 +21,13 @@ def query_ollama(prompt: str, model: str = "mistral"):
 
 @app.get("/")
 async def root():
-    return {"message": "Bienvenue sur mon API avec Ollama"}
+    return {"message": "API avec Ollama sur Hugging Face Spaces"}
 
-# Endpoint pour résumer un texte uploadé
+# Exemple d’endpoint pour résumer un texte
 @app.post("/summarization/text")
 async def summarize_text(file: UploadFile = File(...)):
-    # Lire le contenu du fichier uploadé
     content = await file.read()
     text = content.decode("utf-8")
-    
-    # Créer un prompt pour Ollama
     prompt = f"Résume ce texte en 3 phrases courtes : {text}"
-    
-    # Appeler Ollama avec le modèle (Mistral ou DeepSeek si disponible)
-    summary = query_ollama(prompt, model="mistral")  # Remplace "mistral" par "deepseek-v2" si tu l’as
-    
-    return {"summary": summary}
-
-# Exemple d’endpoint pour tester avec du texte brut
-@app.post("/summarization/test")
-async def summarize_test(text: str):
-    prompt = f"Résume ce texte en 3 phrases courtes : {text}"
-    summary = query_ollama(prompt, model="mistral")
+    summary = query_ollama(prompt, model="mistral")  # Remplace par "deepseek-v2" si dispo
     return {"summary": summary}