Update app.py

app.py

@@ -229,59 +229,7 @@ def process_documents(temp_dir):
     
     return pd.DataFrame(d)
 
-def handle_upload(zip_file, csv_file):
-    """Handle file uploads and process requirements with enhanced error handling."""
-    try:
-        # Create temporary directory
-        temp_dir = tempfile.mkdtemp()
-        print(f"Created temporary directory: {temp_dir}")
-        
-        try:
-            # Extract zip file
-            print(f"Extracting ZIP file: {zip_file.name}")
-            with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
-                zip_ref.extractall(temp_dir)
-                print(f"ZIP contents: {zip_ref.namelist()}")
-            
-            # Preprocess and read requirements CSV
-            print("Processing CSV file...")
-            requirements_df = preprocess_csv(csv_file)
-            print(f"Found {len(requirements_df)} requirements")
-            
-            # Setup RAG system
-            print("Setting up RAG system...")
-            vector_store = setup_rag_system(temp_dir)
-            rag_chain = create_workflow(vector_store)
-            
-            # Process requirements
-            results = []
-            for idx, req in enumerate(requirements_df['requirement'], 1):
-                print(f"Processing requirement {idx}/{len(requirements_df)}")
-                try:
-                    response = rag_chain.invoke(req)
-                    results.append({
-                        'requirement': req,
-                        'response': response
-                    })
-                except Exception as e:
-                    error_msg = f"Error processing requirement: {str(e)}"
-                    print(error_msg)
-                    results.append({
-                        'requirement': req,
-                        'response': error_msg
-                    })
-            
-            return pd.DataFrame(results)
-            
-        finally:
-            # Cleanup
-            print(f"Cleaning up temporary directory: {temp_dir}")
-            shutil.rmtree(temp_dir)
-            
-    except Exception as e:
-        error_msg = f"Processing error: {str(e)}"
-        print(error_msg)
-        return pd.DataFrame([{'error': error_msg}])
+
 
 # The rest of the code remains the same...
 
@@ -387,26 +335,33 @@ def preprocess_csv(csv_file):
             raise ValueError(f"Could not process CSV file: {str(e2)}")
 
 def handle_upload(zip_file, csv_file):
-    """Handle file uploads and process requirements."""
+    """Handle file uploads and process requirements with enhanced error handling."""
     try:
         # Create temporary directory
         temp_dir = tempfile.mkdtemp()
+        print(f"Created temporary directory: {temp_dir}")
         
         try:
             # Extract zip file
+            print(f"Extracting ZIP file: {zip_file.name}")
             with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
                 zip_ref.extractall(temp_dir)
+                print(f"ZIP contents: {zip_ref.namelist()}")
             
             # Preprocess and read requirements CSV
+            print("Processing CSV file...")
             requirements_df = preprocess_csv(csv_file)
+            print(f"Found {len(requirements_df)} requirements")
             
             # Setup RAG system
+            print("Setting up RAG system...")
             vector_store = setup_rag_system(temp_dir)
             rag_chain = create_workflow(vector_store)
             
             # Process requirements
             results = []
-            for req in requirements_df['requirement']:
+            for idx, req in enumerate(requirements_df['requirement'], 1):
+                print(f"Processing requirement {idx}/{len(requirements_df)}")
                 try:
                     response = rag_chain.invoke(req)
                     results.append({
@@ -414,20 +369,24 @@ def handle_upload(zip_file, csv_file):
                         'response': response
                     })
                 except Exception as e:
+                    error_msg = f"Error processing requirement: {str(e)}"
+                    print(error_msg)
                     results.append({
                         'requirement': req,
-                        'response': f"Error processing requirement: {str(e)}"
+                        'response': error_msg
                     })
             
             return pd.DataFrame(results)
             
         finally:
             # Cleanup
+            print(f"Cleaning up temporary directory: {temp_dir}")
             shutil.rmtree(temp_dir)
             
     except Exception as e:
-        return pd.DataFrame([{'error': str(e)}])
-
+        error_msg = f"Processing error: {str(e)}"
+        print(error_msg)
+        return pd.DataFrame([{'error': error_msg}])
 def main():
     """Main function to run the Gradio interface."""
     iface = gr.Interface(