Update app.py

app.py

@@ -740,31 +740,31 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
         logging.error(traceback.format_exc())
         return "Sorry, I encountered an error while processing your request.", []
 
-def generate_answer(message, choice, retrieval_mode, selected_model, selected_file):
-    # Ensure a file is selected
-    if not selected_file:
-        return "Please choose a file to proceed."
-
-    # Modify the Phi-3.5 prompt to include the selected file
-    if selected_model == phi_pipe:
-        retriever = phi_retriever
-        context_documents = retriever.get_relevant_documents(message)
-        context = "\n".join([doc.page_content for doc in context_documents])
-
-        prompt = phi_custom_template.format(context=context, question=message, document_name=selected_file)
-        response = selected_model(prompt, **{
-            "max_new_tokens": 250,
-            "return_full_text": True,
-            "temperature": 0.0,
-            "do_sample": False,
-        })
-
-        if response:
-            generated_text = response[0]['generated_text']
-            cleaned_response = clean_response(generated_text)
-            return cleaned_response
-        else:
-            return "No response generated.", []
+# def generate_answer(message, choice, retrieval_mode, selected_model, selected_file):
+#     # Ensure a file is selected
+#     if not selected_file:
+#         return "Please choose a file to proceed."
+
+#     # Modify the Phi-3.5 prompt to include the selected file
+#     if selected_model == phi_pipe:
+#         retriever = phi_retriever
+#         context_documents = retriever.get_relevant_documents(message)
+#         context = "\n".join([doc.page_content for doc in context_documents])
+
+#         prompt = phi_custom_template.format(context=context, question=message, document_name=selected_file)
+#         response = selected_model(prompt, **{
+#             "max_new_tokens": 250,
+#             "return_full_text": True,
+#             "temperature": 0.0,
+#             "do_sample": False,
+#         })
+
+#         if response:
+#             generated_text = response[0]['generated_text']
+#             cleaned_response = clean_response(generated_text)
+#             return cleaned_response
+#         else:
+#             return "No response generated.", []