Update app.py

app.py

@@ -329,19 +329,19 @@ def chat(question, manual, history, liked, graded_doc_history):
     
     path = "Contents/" + manual + ".json"
     
-    #start_time = time.time()
+    start_time = time.time()
     documents = []
     for i in arg_list:
         documents.append(get_section_content(str(i['section_title']), str(i['sub_section_title']), str(i['sub_sub_section_title']), content_path=path))
 
     CAR_ID = manual_list[manual]
     
-    #documents_db = retriever_db(qclient, question, collection_name, CAR_ID)
+    documents_db = retriever_db(qclient, question, collection_name, CAR_ID)
     
-    #for i in documents_db:
-    #    documents.append(i)
+    for i in documents_db:
+        documents.append(i)
         
-    #print("Retriever:--- %s seconds ---" % (time.time() - start_time))
+    print("Retriever:--- %s seconds ---" % (time.time() - start_time))
 
     start_time = time.time()
     graded_doc = []
@@ -355,24 +355,24 @@ def chat(question, manual, history, liked, graded_doc_history):
 
     print("Onaylanan Doküman Sayısı:", len(graded_doc))
     
-    if not graded_doc:
-        documents = []
-        CAR_ID = manual_list[manual]
-    
-        documents_db = retriever_db(qclient, question, collection_name, CAR_ID)
-        
-        for i in documents_db:
-            documents.append(i)
-        
-        for r in documents:
-        
-            grade = grade_document_with_openai(r, question)
-        
-            if grade.binary_score == "yes":
-                graded_doc.append(r)
-                graded_doc_history.append(r)
-
-        print("Onaylanan Doküman Sayısı DB:", len(graded_doc))
+    #if not graded_doc:
+    #    documents = []
+    #    CAR_ID = manual_list[manual]
+    #
+    #    documents_db = retriever_db(qclient, question, collection_name, CAR_ID)
+    #    
+    #    for i in documents_db:
+    #        documents.append(i)
+    #    
+    #    for r in documents:
+    #    
+    #        grade = grade_document_with_openai(r, question)
+    #    
+    #        if grade.binary_score == "yes":
+    #            graded_doc.append(r)
+    #            graded_doc_history.append(r)
+    #
+    #    print("Onaylanan Doküman Sayısı DB:", len(graded_doc))
     print("Grade Doc:--- %s seconds ---" % (time.time() - start_time))