Update app.py

app.py

@@ -8,6 +8,7 @@ from sentence_transformers import SentenceTransformer
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import json
+import re
 
 DATASET_REPO = "LCA/HACKATHON_PARTS" 
 
@@ -64,6 +65,26 @@ def rechercher_article(articleSource):
    
     return article
 
+def extract_json_from_response(response):
+    """
+    Extrait le JSON d'une chaîne de texte contenant potentiellement du texte en vrac.
+    Retourne un objet Python (dict) ou None si extraction impossible.
+    """
+
+    # Recherche la première accolade ouvrante et fermante pour délimiter le JSON
+    match = re.search(r'({.*})', response, re.DOTALL)
+    if match:
+        json_str = match.group(1)
+        try:
+            return json.loads(json_str)
+        except Exception as e:
+            print("Erreur lors du parsing JSON extrait:", e)
+            return None
+    else:
+        print("Aucun JSON trouvé dans la réponse.")
+        return None
+
+
 def respond(message):
     print(" ------------------ ")
     print(message)
@@ -105,16 +126,18 @@ def respond(message):
 
     print("---- retour de l'analyse")
     print(full_response)
-    print("")
+    print("--")
+    json_response = extract_json_from_response(full_response)
+    print(json_response)
 
     # If you expect a JSON response, you can try to parse it here
     # import json
     # try:
     order = {}
     try:
-        data = json.loads(full_response)
+        
         articles = []
-        for article in data.get("articles", []):
+        for article in json_response.get("articles", []):
             found_article = rechercher_article(article)
             articles.append(found_article)
         order["articles"] = articles