Update app.py

app.py

@@ -69,14 +69,12 @@ def download_gitlab_project_by_version():
         response = requests.get(url, stream=True)
         archive_bytes = io.BytesIO(response.content)
 
-        print(archive_bytes)
         
         if response.status_code == 200:
             # Extract filename from content-disposition header
             content_disposition = response.headers.get("content-disposition")
             if content_disposition and "filename=" in content_disposition:
                 filename = content_disposition.split("filename=")[-1].strip('"')
-        print(filename)
 
 
         # test
@@ -150,9 +148,7 @@ def download_gitlab_repo():
 def get_all_files_in_folder(temp_dir, folder_path):
     
     all_files = [] 
-    print("inner method of get all files in folder")
     target_dir = os.path.join(temp_dir, folder_path)
-    print(target_dir)
 
     for root, dirs, files in os.walk(target_dir):
         print(f"Files in current directory ({root}): {files}")
@@ -171,11 +167,8 @@ def process_directory(directory, folder_paths, file_paths):
     all_texts = []
     file_references = []
 
-    zip_filename = next((file for file in os.listdir(directory) if file.endswith('.zip')), None) # zip_filename: kadi-apy-master-2a244f1af1483b48f8f9c0d99ce2744a0950c834.zip
-    print("zip_filename:", zip_filename)
-    zip_file_path = os.path.join(directory, zip_filename)  # zip_file_path: data/kadi-apy-master-2a244f1af1483b48f8f9c0d99ce2744a0950c834.zip
-    print("zip_file_path:", zip_file_path)
-   # zip_file_path = os.listdir(directory) if file.endswith('.zip')
+    zip_filename = next((file for file in os.listdir(directory) if file.endswith('.zip')), None)
+    zip_file_path = os.path.join(directory, zip_filename)
 
     with tempfile.TemporaryDirectory() as tmpdirname:
         # Unzip the file into the temporary directory
@@ -183,11 +176,11 @@ def process_directory(directory, folder_paths, file_paths):
             zip_ref.extractall(tmpdirname)
             
             files = []
-            print("tmpdirname: " , tmpdirname)     # /tmp/tmpux1v52wy
+            print("tmpdirname: " , tmpdirname)   
             unzipped_root = os.listdir(tmpdirname)
-            print("unzipped_root ", unzipped_root)  # ['kadi-apy-master-2a244f1af1483b48f8f9c0d99ce2744a0950c834']
+            print("unzipped_root ", unzipped_root)
 
-            tmpsubdirpath= os.path.join(tmpdirname, unzipped_root[0])  # /tmp/tmpux1v52wy/kadi-apy-master-2a244f1af1483b48f8f9c0d99ce2744a0950c834
+            tmpsubdirpath= os.path.join(tmpdirname, unzipped_root[0])  
             print("tempsubdirpath: ", tmpsubdirpath)
 
             if folder_paths:
@@ -216,7 +209,6 @@ def process_directory(directory, folder_paths, file_paths):
                     relative_path = os.path.relpath(file_path, tmpsubdirpath)
                     print("Relative Filepaths brother:", relative_path)
                     file_references.append(relative_path)
-                    print("AAAAAAAAAAAAAAAAAAAAAAAAAAAAA: ", relative_path)
                     
     return all_texts, file_references
                 
@@ -312,6 +304,7 @@ def format_kadi_apy_library_context(docs):
         type_info = doc.metadata.get("type", "Unknown Type")
         source_info = doc.metadata.get("source", "Unknown Type")
         # Format metadata and document content
+        print("YYYYYYYEEEEEEEEEEEEEEE222222222222222222222222222222:}\n\n", doc.page_content)
         formatted_doc = f"# source: {source_info}\n# class: {class_info}\n# type: {type_info}\n{doc.page_content}\n\n\n"
         doc_context.append(formatted_doc)
     
@@ -342,9 +335,9 @@ def rag_workflow(query):
     kadi_apy_library_docs = retrieve_within_kadiApy_library (vector_store, query, k = 10)
     
     doc_context = format_kadi_api_doc_context(kadi_apy_docs)
-    print(doc_context)
+    #print(doc_context)
     code_context = format_kadi_apy_library_context(kadi_apy_library_docs)
-    print("AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA:",code_context)
+    #print("AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA:",code_context)