Update space

app.py

@@ -1,35 +1,35 @@
 import gradio as gr
 import requests
+import json
 
-# Fonction pour interroger Ollama
 def query_ollama(prompt):
-    # Configuration de l'URL et du modèle Ollama
-    url = "http://localhost:11434/api/generate"  # Le port utilisé par Ollama
-    model_name = "hf.co/ibrahimBlyc/LA_Llama:latest"  # Nom complet du modèle
-    
-    # Corps de la requête
+    url = "http://localhost:11434/api/generate"
     payload = {
-        "model": model_name,
+        "model": "hf.co/ibrahimBlyc/LA_Llama:latest",
         "prompt": prompt
     }
-    
-    try:
-        # Envoi de la requête POST
-        response = requests.post(url, json=payload)
-        response.raise_for_status()  # Vérifie les erreurs HTTP
-        result = response.json().get("response", "Aucune réponse reçue.")
-    except requests.exceptions.RequestException as e:
-        result = f"Erreur : {e}"
-    
-    return result
+    headers = {"Content-Type": "application/json"}
+    response = requests.post(url, json=payload, headers=headers, stream=True)
+
+    if response.status_code == 200:
+        full_response = ""
+        for line in response.iter_lines():
+            if line:
+                data = json.loads(line.decode("utf-8"))
+                if "response" in data:
+                    full_response += data["response"]
+                if data.get("done", False):
+                    break
+        return full_response
+    else:
+        return f"Erreur HTTP : {response.status_code}"
 
-# Interface Gradio
 interface = gr.Interface(
     fn=query_ollama,
-    inputs=gr.Textbox(label="Entrée utilisateur", placeholder="Posez une question..."),
-    outputs=gr.Textbox(label="Réponse du modèle"),
-    title="Interface Ollama via Hugging Face Spaces",
-    description="Une interface simple pour communiquer avec un modèle Ollama via le port 11434."
+    inputs=gr.Textbox(label="Posez votre question"),
+    outputs=gr.Textbox(label="Réponse"),
+    title="Chat avec Ollama",
+    description="Une interface simple pour discuter avec le modèle hébergé sur Ollama."
 )
 
 if __name__ == "__main__":