Update app.py

app.py

@@ -10,17 +10,15 @@ import re
 import shutil
 import glob
 def delete_files_except(filename, folder_path):
+    delfile=None
     files = glob.glob(os.path.join(folder_path, '*'))
     for file in files:
         if os.path.basename(file) == filename:
             continue 
         else:
-            try:
-                os.remove(file) 
-                print(f"Deleted: {file}")
-            except Exception as e:
-                print(f"Error deleting {file}: {e}")
-
+            delfile=os.path.basename(file)
+            os.remove(file) 
+    return delfile
 def MOP(path):
   docs=[]
   loader = PyPDFDirectoryLoader(path)
@@ -30,15 +28,15 @@ def MOP(path):
   embeddings = HuggingFaceEmbeddings(model_name="thenlper/gte-base")
   vector_store = FAISS.from_documents(text_chunks, embedding=embeddings)
   filename_to_keep = 'Dummy_standard MoP_template_new.pdf'
-  delete_files_except(filename_to_keep,path)
+  prompt_file=delete_files_except(filename_to_keep,path)
   repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
   llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 2048})
   retriever = vector_store.as_retriever(search_type="similarity",search_kwargs={"k": len(text_chunks)})
   qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever,verbose=True)
-  message= """<s> [INST] You have two documents:
+  message= f"""<s> [INST] You have two documents:
 
 Template Document: "Dummy_standard MoP_template_new.pdf"
-Content Document: "Fully_Standard_MOP_of_Activation_Carrier_Aggregaton_in_v1.pdf"
+Content Document: {prompt_file}
 
 Your task is to determine whether the Content Document effectively incorporates the context and instructions of the Pre-Check Procedures and Post-Check Procedures specified in the Template Document, without the requirement for them to be under specified labels. The focus should be on the presence of the context and instructions rather than their exact placement.
 
@@ -49,10 +47,10 @@ Additionally, identify any missing elements related to the context and instructi
 Instructions:
 
 Review the context and instructions of the Pre-Check Procedures and Post-Check Procedures detailed in the Template Document ("Dummy_standard MoP_template_new.pdf").
-Assess whether the Content Document ("Fully_Standard_MOP_of_Activation_Carrier_Aggregaton_in_v1.pdf") includes the necessary context and instructions for the Pre-Check Procedures and Post-Check Procedures, regardless of their specific placement or labeling.
+Assess whether the Content Document ({prompt_file}) includes the necessary context and instructions for the Pre-Check Procedures and Post-Check Procedures, regardless of their specific placement or labeling.
 Provide a "Yes" if the Content Document adequately integrates the context and instructions of the Pre-Check Procedures and Post-Check Procedures, or "No" if there are significant gaps or omissions.
 If the answer is "No," specify any missing elements related to the context and instructions of the Pre-Check Procedures and Post-Check Procedures, emphasizing their importance in the Content Document.
-Ensure careful consideration of the context and instructions provided in the Template Document ("Dummy_standard MoP_template_new.pdf") while evaluating the alignment of the Content Document ("Fully_Standard_MOP_of_Activation_Carrier_Aggregaton_in_v1.pdf"). [/INST] </s>"""
+Ensure careful consideration of the context and instructions provided in the Template Document ("Dummy_standard MoP_template_new.pdf") while evaluating the alignment of the Content Document ({prompt_file}). [/INST] </s>"""
   result=qa.run(message)
   pattern = r"Helpful Answer:\n\n(.*)"
   match = re.search(pattern, result, re.DOTALL)