use filename as column

app.py

@@ -88,7 +88,7 @@ def wake_servers(progress=gr.Progress()):
     
     def try_wake_endpoint(name, url):
         """Helper function to wake endpoint with retry logic"""
-        retry_delays = [30, 20, 20, 20, 20, 20, 20, 20, 20, 20]  # Seconds to wait between retries
+        retry_delays = [30, 30, 30, 30, 30, 30, 30, 30, 30, 30]  # Seconds to wait between retries
         
         for retry_count, retry_delay in enumerate(retry_delays):
             try:
@@ -297,10 +297,14 @@ def process_resume(resume_text: str, job_skills: List[str], progress=None, progr
     
     return formatted_result
 
-def create_html_output(job_result: Dict, resume_results: List[Dict]) -> str:
+def create_html_output(job_result: Dict, resume_results: List[Dict], filenames: List[str] = None) -> str:
     """Create HTML output for the interface"""
     html = "<div style='font-family: Arial, sans-serif;'>"
     
+    # Set default filenames if not provided
+    if not filenames:
+        filenames = [f"Resume {i}" for i in range(1, len(resume_results) + 1)]
+    
     # Job Description Section
     html += "<h2>Job Description Analysis</h2>"
     html += f"<p><strong>Total Skills Found:</strong> {job_result['total_skills']}</p>"
@@ -449,7 +453,8 @@ def create_html_output(job_result: Dict, resume_results: List[Dict]) -> str:
                 'total_experience': total_experience,
                 'category': category,
                 'quality_score': quality_score,
-                'resume_result': resume_result
+                'resume_result': resume_result,
+                'filename': filenames[i-1]  # Store the filename
             })
         
         # Sort resume data:
@@ -500,9 +505,10 @@ def create_html_output(job_result: Dict, resume_results: List[Dict]) -> str:
             collaboration_count = resume_data['collaboration_count']
             total_experience = resume_data['total_experience']
             category = resume_data['category']
+            filename = resume_data['filename']  # Get the filename
             
-            # Add row to summary table
-            html += f"<tr><td style='padding: 8px; border: 1px solid #ddd;'>Resume {i}</td>"
+            # Add row to summary table - using filename instead of "Resume {i}"
+            html += f"<tr><td style='padding: 8px; border: 1px solid #ddd;'>{filename}</td>"
             html += f"<td style='padding: 8px; text-align: center; border: 1px solid #ddd;'>{match_count}/{job_result['total_skills']}</td>"
             html += f"<td style='padding: 8px; text-align: center; border: 1px solid #ddd;'>{match_percentage}%</td>"
             html += f"<td style='padding: 8px; text-align: center; border: 1px solid #ddd;'>{leadership_count}</td>"
@@ -552,9 +558,12 @@ def create_html_output(job_result: Dict, resume_results: List[Dict]) -> str:
             html += f"<p><strong>Total Years of Experience:</strong> {total_experience}</p>"
             html += f"<p><strong>Category:</strong> {category}</p>"
         
+        # Get the filename for this resume
+        resume_file = filenames[i-1]
+        
         # Detailed resume section with visibility control
         html += f"<div id='resume-detail-{i}' class='resume-detail' style='margin-bottom: 20px; padding: 10px; border: 1px solid #ddd; border-radius: 5px; display: {display_style};'>"
-        html += f"<h3>Resume {i} Details</h3>"
+        html += f"<h3>{resume_file} Details</h3>"
         
         # Display all skills found in the resume
         html += "<p><strong>Skills Found:</strong></p>"
@@ -643,8 +652,11 @@ def process_single_resume(file_path, job_skills, progress=None, resume_index=0,
     with open(file_path, 'r', encoding='utf-8') as f:
         resume_content = f.read()
     
-    return process_resume(resume_content, job_skills, 
+    result = process_resume(resume_content, job_skills, 
                         progress=progress, progress_base=progress_base, progress_cap=progress_cap)
+    
+    # Return both the result and the filename
+    return result, os.path.basename(file_path)
 
 def process_inputs(job_description: str, input_type: str, resume_text: str, resume_files: List[str], progress=gr.Progress()) -> str:
     """Main processing function"""
@@ -657,12 +669,15 @@ def process_inputs(job_description: str, input_type: str, resume_text: str, resu
     
     # Process resumes based on input type
     resume_results = []
+    filenames = []  # Track filenames for multiple uploads
+    
     if input_type == "Paste Text":
         # Process single resume from text input
         progress(0.4, desc="Analyzing resume structure and extracting roles...")
         resume_result = process_resume(resume_text, job_skills, 
                                       progress=progress, progress_base=0.4, progress_cap=0.9)
         resume_results.append(resume_result)
+        filenames.append("Pasted Resume")  # Default name for pasted text
     else:
         # Process multiple resumes from file uploads in parallel
         resume_count = len(resume_files)
@@ -678,14 +693,17 @@ def process_inputs(job_description: str, input_type: str, resume_text: str, resu
                     progress, 
                     i, 
                     resume_count
-                ): i for i, file_path in enumerate(resume_files)
+                ): file_path for i, file_path in enumerate(resume_files)
             }
             
             # Collect results as they complete
             completed = 0
             total = len(future_to_resume)
             for future in concurrent.futures.as_completed(future_to_resume):
-                resume_results.append(future.result())
+                file_path = future_to_resume[future]
+                result, filename = future.result()
+                resume_results.append(result)
+                filenames.append(filename)
                 completed += 1
                 try:
                     if progress is not None:
@@ -696,7 +714,7 @@ def process_inputs(job_description: str, input_type: str, resume_text: str, resu
     
     # Create HTML output
     progress(0.9, desc="Generating results and preparing final report...")
-    html_output = create_html_output(job_result, resume_results)
+    html_output = create_html_output(job_result, resume_results, filenames)
     progress(1.0, desc="Analysis complete!")
     return html_output