Update app.py

app.py

@@ -475,7 +475,6 @@ def main():
                         history[-1][1] += html_escape(str(character))
                         time.sleep(0.05)
                         yield history,responseHTML   
-     
 
             else:
                 print("MODALITA STANDARD")
@@ -538,11 +537,18 @@ def main():
                         retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.5)
                        
                 if(str(current_response_mode)=="tree_summarize"):
+                    # define response synthesizer
+                    response_synthesizer = get_response_synthesizer(streaming=True,response_mode="tree_summarize",text_qa_template=text_qa_template)
+                    query_engine = None
+                    query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
+                    stream_response = None
+                    print(userMessage)
                     # Misura il tempo di inizio
                     start_time = time.time()
                     
                     # Esegui la funzione
-                    retriever.retrieve(userMessage)
+                    stream_response = query_engine.query(userMessage)
+                    
                     
                     # Misura il tempo di fine
                     end_time = time.time()
@@ -550,14 +556,10 @@ def main():
                     # Calcola il tempo di esecuzione
                     execution_time = end_time - start_time
                     print(f"Tempo di esecuzione: {execution_time} secondi")
-                    # define response synthesizer
-                    response_synthesizer = get_response_synthesizer(streaming=True,response_mode="tree_summarize",text_qa_template=text_qa_template)
-                    query_engine = None
-                    query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
-                    stream_response = None
-                    print(userMessage)
-                    stream_response = query_engine.query(userMessage)
+                    #stream_response = query_engine.query(userMessage)
                     print("risposta con query engine")
+
+                    
                     responseHTML = ""
                     for i, node in enumerate(stream_response.source_nodes):
                         responseHTML += "<p><b>"+node.metadata['nome_bando']+"</b><a href='"+node.metadata['file_path']+"' download> <svg xmlns='http://www.w3.org/2000/svg' width='16' height='16' fill='currentColor' class='bi bi-download' viewBox='0 0 16 16'><path d='M.5 9.9a.5.5 0 0 1 .5.5v2.5a1 1 0 0 0 1 1h12a1 1 0 0 0 1-1v-2.5a.5.5 0 0 1 1 0v2.5a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2v-2.5a.5.5 0 0 1 .5-.5'/><path d='M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708z'/> </svg></a><br>Nodo <span id='span"+str(i+1)+"' class='icon' onclick='toggleText(this.id)'>🔍</span>   <!-- Text to show/hide --><p class='hidden-text' id='nodo"+str(i+1)+"'>"+node.text+"</p>"