Update app.py

app.py

@@ -112,24 +112,24 @@ css = """
 
 """
 
-user_message = ""
-current_chat_mode = ""
-current_response_mode = "compact"
-current_collection = ""
-file_path = ""
-num_responses = 0
-current_chat_mode = "STANDARD"
-retriever = None
-token_count_bandi = 0
-token_count_bandi_sistema_puglia = 0
-chat_engine_bandi = None
-chat_engine_bandi_sistema_puglia = None
-memory_bandi = None
-memory_bandi_sistema_puglia = None
-stream_response = None
-divDocumenti = None
-
 def main():
+    user_message = ""
+    current_chat_mode = ""
+    current_response_mode = "compact"
+    current_collection = ""
+    file_path = ""
+    num_responses = 0
+    current_chat_mode = "STANDARD"
+    retriever = None
+    token_count_bandi = 0
+    token_count_bandi_sistema_puglia = 0
+    chat_engine_bandi = None
+    chat_engine_bandi_sistema_puglia = None
+    memory_bandi = None
+    memory_bandi_sistema_puglia = None
+    stream_response = None
+    divDocumenti = None
+
     setGPU()
     llm = setLLM()
     Settings.llm = llm
@@ -419,6 +419,7 @@ def main():
                     print(chat_engine_bandi.chat_history)
                     print(memory_bandi)
                     stream_response = None
+                    print(user_message)
                     stream_response = chat_engine_bandi.stream_chat(user_message)
                     print("risposta con chat engine")
                     responseHTML = ""
@@ -444,6 +445,7 @@ def main():
                     print(chat_engine_bandi_sistema_puglia.chat_history)
                     print(memory_bandi_sistema_puglia)
                     stream_response = None
+                    print(user_message)
                     stream_response = chat_engine_bandi_sistema_puglia.stream_chat(user_message)
                     print("risposta con chat engine")
                     responseHTML = "" 
@@ -469,6 +471,7 @@ def main():
                     query_engine = None
                     query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
                     stream_response = None
+                    print(user_message)
                     stream_response = query_engine.query(user_message)
                     print("risposta con query engine")
                     responseHTML = ""
@@ -486,6 +489,7 @@ def main():
                     query_engine = None
                     query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
                     stream_response = None
+                    print(user_message)
                     stream_response = query_engine.query(user_message)
                     print("risposta con query engine")
                     responseHTML = ""
@@ -504,7 +508,8 @@ def main():
             torch.cuda.empty_cache()
             torch.cuda.reset_max_memory_allocated()
             torch.cuda.reset_max_memory_cached()
-
+            
+            
 
         msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
             bot, chatbot, [chatbot, divDocumenti]