Update app.py

app.py

@@ -613,7 +613,9 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
 
         if retrieval_mode == "VDB":
             logging.debug("Using VDB retrieval mode")
-            if selected_model in [gpt_model, gpt_mini_model]:
+            # Adjust this block to handle both LM-1 and LM-3
+            if selected_model in [gpt_model, gpt_mini_model]:  # Both LM-1 and LM-3 should use the same logic
+                logging.debug(f"Selected model: {'LM-1' if selected_model == gpt_model else 'LM-3'}")
                 retriever = gpt_retriever
                 context = retriever.get_relevant_documents(message)
                 logging.debug(f"Retrieved context: {context}")
@@ -631,13 +633,16 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
                 logging.debug(f"LM-1 or LM-3 response: {response}")
                 return response['result'], extract_addresses(response['result'])
 
-            elif selected_model == phi_pipe:
+            elif selected_model == phi_pipe:  # LM-2 specific logic
+                logging.debug("Selected model: LM-2")
                 retriever = phi_retriever
                 context_documents = retriever.get_relevant_documents(message)
                 context = "\n".join([doc.page_content for doc in context_documents])
                 logging.debug(f"Retrieved context for LM-2: {context}")
 
                 prompt = phi_custom_template.format(context=context, question=message)
+                logging.debug(f"Generated LM-2 prompt: {prompt}")
+
                 response = selected_model(prompt, **{
                     "max_new_tokens": 400,
                     "return_full_text": True,
@@ -647,6 +652,7 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
 
                 if response:
                     generated_text = response[0]['generated_text']
+                    logging.debug(f"LM-2 Response: {generated_text}")
                     cleaned_response = clean_response(generated_text)
                     return cleaned_response, extract_addresses(cleaned_response)
                 else:
@@ -654,7 +660,9 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
                     return "No response generated.", []
 
         elif retrieval_mode == "KGF":
+            logging.debug("Using KGF retrieval mode")
             response = chain_neo4j.invoke({"question": message})
+            logging.debug(f"KGF response: {response}")
             return response, extract_addresses(response)
         else:
             logging.error("Invalid retrieval mode selected.")
@@ -668,6 +676,7 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
 
 
 
+
 def add_message(history, message):
     history.append((message, None))
     return history, gr.Textbox(value="", interactive=True, show_label=False)
@@ -1141,16 +1150,28 @@ def handle_retrieval_mode_change(choice):
 
 
 
+# def handle_model_choice_change(selected_model):
+#     if selected_model == "LM-2":
+#         # Disable retrieval mode and select style when LM-2 is selected
+#         return gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)
+#     elif selected_model == "LM-1":
+#         # Enable retrieval mode and select style for LM-1
+#         return gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
+#     else:
+#         # Default case: allow interaction
+#         return gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
+
+
 def handle_model_choice_change(selected_model):
     if selected_model == "LM-2":
         # Disable retrieval mode and select style when LM-2 is selected
         return gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)
-    elif selected_model == "LM-1":
-        # Enable retrieval mode and select style for LM-1
+    elif selected_model in ["LM-1", "LM-3"]:
+        # Enable retrieval mode and select style for LM-1 and LM-3
         return gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
     else:
         # Default case: allow interaction
-        return gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
+        return gr.update(interactive=True), gr.update(interactive(True), gr.update(interactive=True)