Update app.py

app.py

@@ -225,15 +225,29 @@ def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Pr
     progress(0.9, desc="Done!")
     return vector_db, collection_name, "Complete!"
 
-
-def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
-    # print("llm_option",llm_option)
+def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress(), language=None):
+    # Definiamo le lingue supportate
+    supported_languages = ['it', 'en']  # Italiano e inglese
+    
+    # Se la lingua non è specificata, usiamo l'italiano come default
+    if language is None:
+        language = 'it'
+    
+    # Verifichiamo se la lingua è supportata
+    if language not in supported_languages:
+        raise ValueError(f"Lingua non supportata. Supportati: {supported_languages}")
+    
+    # Convertiamo la lingua in minuscolo per la compatibilità con il modello
+    language = language.lower()
+    
     llm_name = list_llm[llm_option]
-    print("llm_name: ",llm_name)
-    qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, progress)
+    print(f"Nome del modello: {llm_name}")
+    
+    # Creiamo il chain LL.M. con la lingua specificata
+    qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, progress, language)
+    
     return qa_chain, "Complete!"
 
-
 def format_chat_history(message, chat_history):
     formatted_chat_history = []
     for user_message, bot_message in chat_history: