Update main.py

main.py

@@ -46,6 +46,8 @@ def filtergpt(text):
     gpt_results = [(law, str(int(article)) if article.is_integer() else str(article)) for law, article in law_article_list]
     return gpt_results
 
+import ast
+
 def perform_search_and_get_results(collection_name, query, limit=6):
     search_results = client.search(
         collection_name=collection_name,
@@ -59,9 +61,11 @@ def perform_search_and_get_results(collection_name, query, limit=6):
             "La_loi": result.payload["reference"],
             "Paragraphe": result.payload["paragraph"],
             "titre": result.payload["titre"],
+            "section_text": result.payload["section"],
+            "section_label": result.payload["section_label"],
             "source": result.payload["source"],
             "collection": collection_name,
-            "hyperlink": result.payload["hyperlink"],
+            "hyperlink": ast.literal_eval(result.payload['hyperlink']),
         }
         resultes.append(result_dict)
     return resultes
@@ -79,11 +83,12 @@ def perform_search_and_get_results_with_filter(collection_name, query,reference_
             "Score": result.score,
             "La_loi": result.payload["reference"],
             "Paragraphe": result.payload["paragraph"],
-            "source": result.payload["source"],
             "titre": result.payload["titre"],
+            "section_text": result.payload["section"],
+            "section_label": result.payload["section_label"],
+            "source": result.payload["source"],
             "collection": collection_name,
-            "hyperlink": result.payload["hyperlink"],
-
+            "hyperlink": ast.literal_eval(result.payload['hyperlink']),
         }
         resultes.append(result_dict)
     return resultes