Update app.py

app.py

@@ -21,10 +21,8 @@ except OSError:
     os.system("python -m spacy download en_core_web_sm")
     nlp = spacy.load("en_core_web_sm")
 
-# Global variables
 generated_summaries = {}
 
-# Regulatory data
 RISK_KEYWORDS = [
     "penalty", "breach", "liability", "default", "hidden obligations",
     "indemnity", "terms of service", "non-compliance", "force majeure"
@@ -207,23 +205,29 @@ Please find the attached PDF summary for your review.
     except Exception as e:
         return f"Failed to send email: {str(e)}"
 
-def process_files(files, tasks):
-    if not files or not tasks:
+def process_uploaded_files(file_obj_list, tasks):
+    if not file_obj_list or not tasks:
         return "Please upload files and select tasks."
     
     results = {}
     all_text_chunks = {}
     
-    for file in files:
-        file_name = os.path.basename(file.name)
+    for file_obj in file_obj_list:
+        file_name = os.path.basename(file_obj.name)
         file_extension = os.path.splitext(file_name)[1].lower()
         
+        # Create a temporary file
+        temp_file_path = f"temp_{file_name}"
+        with open(temp_file_path, "wb") as temp_file:
+            temp_file.write(file_obj.read())
+        
         if file_extension == '.pdf':
-            text = extract_text_from_pdf(file.name)
+            text = extract_text_from_pdf(temp_file_path)
         elif file_extension == '.txt':
-            text = extract_text_from_txt(file.name)
+            text = extract_text_from_txt(temp_file_path)
         else:
-            results[file_name] = "Unsupported file format"
+            results[file_name] = {"Error": "Unsupported file format"}
+            os.remove(temp_file_path)
             continue
             
         text_chunks = split_text_into_chunks(text)
@@ -243,6 +247,9 @@ def process_files(files, tasks):
         results[file_name] = file_results
         generated_summaries[file_name] = file_results
         
+        # Clean up
+        os.remove(temp_file_path)
+        
     return results, all_text_chunks
 
 def display_results(results):
@@ -282,6 +289,38 @@ def answer_questions(file_name, question):
     else:
         return "No summary available for the selected file."
 
+def process_with_progress(file_obj_list, tasks):
+    if not file_obj_list or not tasks:
+        return None, "Please upload files and select tasks."
+        
+    fig, ax = plt.subplots(figsize=(8, 4))
+    ax.set_title("Processing Progress")
+    ax.set_xlabel("Steps")
+    ax.set_ylabel("Progress (%)")
+    ax.grid(True)
+    
+    progress_data = [0]
+    ax.plot(progress_data, color="blue", marker="o")
+    
+    # Process files
+    results, all_text_chunks = process_uploaded_files(file_obj_list, tasks)
+    
+    # Update progress to 100%
+    progress_data.append(100)
+    ax.clear()
+    ax.plot(progress_data, color="blue", marker="o")
+    ax.set_title("Processing Progress")
+    ax.set_xlabel("Steps")
+    ax.set_ylabel("Progress (%)")
+    ax.set_ylim(0, 100)
+    ax.grid(True)
+    
+    # Generate options for dropdowns
+    file_task_choices = [f"{file_name} - {task}" for file_name, file_results in results.items() for task in file_results.keys()]
+    file_choices = list(results.keys())
+    
+    return fig, display_results(results), file_task_choices, file_choices
+
 with gr.Blocks(title="AI-Driven Legal Document Analysis") as app:
     gr.Markdown("""
     # 📜 Advanced AI-Driven Legal Document Summarization and Risk Assessment
@@ -298,7 +337,8 @@ with gr.Blocks(title="AI-Driven Legal Document Analysis") as app:
     with gr.Tab("Process Documents"):
         with gr.Row():
             with gr.Column():
-                files = gr.File(label="Upload PDFs or Text Files", file_types=["pdf", "txt"], multiple=True)
+                # Fixed: Using File() without 'multiple' parameter
+                files = gr.File(label="Upload PDFs or Text Files", file_types=["pdf", "txt"])
                 task_checkboxes = gr.CheckboxGroup(
                     ["Summarize", "Extract Clauses", "Risk Detection", "Regulatory Update Tracker"],
                     label="Choose Tasks"
@@ -312,7 +352,7 @@ with gr.Blocks(title="AI-Driven Legal Document Analysis") as app:
     with gr.Tab("Email Summary"):
         with gr.Row():
             with gr.Column():
-                file_task_dropdown = gr.Dropdown(label="Select a Task Summary to Send", choices=[])
+                file_task_dropdown = gr.Dropdown(label="Select a Task Summary to Send")
                 receiver_email = gr.Textbox(label="Receiver Email")
                 email_subject = gr.Textbox(label="Email Subject", value="Legal Document Summary")
                 sender_email = gr.Textbox(label="Sender Email (Gmail)", value="your_email@gmail.com")
@@ -325,49 +365,16 @@ with gr.Blocks(title="AI-Driven Legal Document Analysis") as app:
     with gr.Tab("Ask Questions"):
         with gr.Row():
             with gr.Column():
-                file_dropdown = gr.Dropdown(label="Select a File Summary", choices=[])
+                file_dropdown = gr.Dropdown(label="Select a File Summary")
                 question = gr.Textbox(label="Ask a Question")
                 ask_btn = gr.Button("Get Answer")
             
             with gr.Column():
                 answer = gr.Textbox(label="Answer")
     
-    def process_with_progress(files, tasks):
-        if not files or not tasks:
-            return None, "Please upload files and select tasks."
-            
-        # Create progress visualization
-        progress_data = []
-        fig, ax = plt.subplots(figsize=(8, 4))
-        ax.set_title("Processing Progress")
-        ax.set_xlabel("Steps")
-        ax.set_ylabel("Progress (%)")
-        ax.grid(True)
-        
-        total_steps = len(files) * len(tasks)
-        current_step = 0
-        
-        results, all_text_chunks = process_files(files, tasks)
+    def update_dropdowns(file_task_choices, file_choices):
+        return gr.Dropdown.update(choices=file_task_choices), gr.Dropdown.update(choices=file_choices)
         
-        file_task_choices = [f"{file_name} - {task}" for file_name, file_results in results.items() for task in file_results.keys()]
-        file_choices = list(results.keys())
-        
-        # For each step, update progress
-        for i in range(total_steps):
-            current_step += 1
-            progress = (current_step / total_steps) * 100
-            progress_data.append(progress)
-            
-            ax.clear()
-            ax.plot(progress_data, color="blue", marker="o")
-            ax.set_title("Processing Progress")
-            ax.set_xlabel("Steps")
-            ax.set_ylabel("Progress (%)")
-            ax.set_ylim(0, 100)
-            ax.grid(True)
-            
-            yield fig, display_results(results), gr.Dropdown.update(choices=file_task_choices), gr.Dropdown.update(choices=file_choices)
-    
     process_btn.click(
         process_with_progress,
         inputs=[files, task_checkboxes],