Edit output (add list sources)

main.py

@@ -163,21 +163,28 @@ async def process_court_decision(url, question, progress=gr.Progress()):
         unique_citations = sorted(set(citations), key=int)
 
         # Prepare output
-        output = f"**Правова позиція:**\n{legal_position_json['Title']}: {legal_position_json['Legal_position']}\n\n"
+        output = f"**Правова позиція нового судового рішення:**\n{legal_position_json['Title']}: {legal_position_json['Legal_position']}\n\n"
         output += f"**Відповідь ШІ:**\n{result.response}\n\n"
 
-        output += "**Цитовані джерела:**\n"
-        for citation in unique_citations:
-            citation_index = int(citation) - 1
-            if 0 <= citation_index < len(result.source_nodes):
-                output += f"[{citation}]: {result.source_nodes[citation_index].node.metadata['title']}\n"
+        output += "**Цитовані джерела існуючих правових позицій Верховного Суду:**\n"
+        # for citation in unique_citations:
+        #     citation_index = int(citation) - 1
+        #     if 0 <= citation_index < len(result.source_nodes):
+        #         output += f"[{citation}]: {result.source_nodes[citation_index].node.metadata['title']}\n"
+
+        # Display each source node with its title and score in Gradio
+        sources_output = ""
+        for index, node in enumerate(result.source_nodes, start=1):
+            sources_output += f"[{index}] {node.metadata['title']} 👉 {node.get_score()} \n\n"
+
+        # Combine main output and sources output
+        final_output = output + "\n\n" + sources_output
 
         progress(1.0, desc="Complete!")
-        return output
+        return final_output
     except Exception as e:
         return f"Error processing court decision: {str(e)}"
 
-
 # Also, add the extract_court_decision_text function if it's not already there
 def extract_court_decision_text(url):
     response = requests.get(url)