Update app.py

app.py

@@ -77,24 +77,34 @@ def pdf_to_text(text, PDF, min_length=20):
 
 # Function to process the preloaded document
 def process_sample_document(min_length=20):
-    # Replace this with the path to your preloaded legal document (PDF or DOCX)
-    sample_document_path = "Marbury v. Madison.pdf"  # Replace with your file's path
+    # Replace this with the path to your preloaded legal document (PDF)
+    sample_document_path = "Marbury v. Madison.pdf"
     
-    # Use the same processing function to summarize the preloaded document
-    return pdf_to_text("", sample_document_path, min_length)
+    # Simulate file-like object for Gradio
+    with open(sample_document_path, "rb") as f:
+        return pdf_to_text("", f, min_length)
 
 # Gradio interface
-iface = gr.Interface(
-    fn=pdf_to_text,
-    inputs=[gr.Textbox(label="Input Text"), gr.File(label="Upload PDF or DOCX"), gr.Slider(minimum=10, maximum=100, step=10, value=20, label="Summary Minimum Length")],
-    outputs=[gr.Audio(label="Generated Audio"), gr.Textbox(label="Generated Summary"), gr.File(label="Summary PDF")],
-    title="Legal Document Summarizer",
-    description="Upload a legal document in PDF or DOCX format to generate a summary.",
-)
-
-# Add a button for the sample document
-iface.add_component(gr.Button("Process Sample Legal Document", elem_id="process_sample_document", interactive=True))
-iface.set_event_listener("process_sample_document", process_sample_document)
+with gr.Blocks() as iface:
+    # Create a button to process the sample document
+    with gr.Row():
+        process_sample_button = gr.Button("Process Sample Legal Document")
+    
+    # Create the regular input interface
+    text_input = gr.Textbox(label="Input Text")
+    file_input = gr.File(label="Upload PDF or DOCX")
+    slider = gr.Slider(minimum=10, maximum=100, step=10, value=20, label="Summary Minimum Length")
+    
+    # Outputs
+    audio_output = gr.Audio(label="Generated Audio")
+    summary_output = gr.Textbox(label="Generated Summary")
+    pdf_output = gr.File(label="Summary PDF")
+    
+    # Define the action for the button click to process the preloaded document
+    process_sample_button.click(fn=process_sample_document, inputs=slider, outputs=[audio_output, summary_output, pdf_output])
+    
+    # Define the action for the regular file processing
+    file_input.change(fn=pdf_to_text, inputs=[text_input, file_input, slider], outputs=[audio_output, summary_output, pdf_output])
 
 if __name__ == "__main__":
     iface.launch()