Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 import gradio as gr
 from transformers import T5ForConditionalGeneration, T5Tokenizer
-import fitz 
+import fitz  
 
 # Load the model and tokenizer
 model_path = 'Sibinraj/T5-finetuned-dialogue_sumxx'
@@ -47,7 +47,7 @@ def summarize_text(text, max_length, show_length):
     summary_ids = model.generate(
         inputs,
         max_length=max_length + 20,  # Allow some buffer
-        min_length=10,  
+        min_length=10,  # Set a reasonable minimum length
         num_beams=5,
         no_repeat_ngram_size=2,
         early_stopping=True
@@ -76,11 +76,13 @@ def summarize_text(text, max_length, show_length):
     
     return summary
 
-def handle_pdf(pdf, max_length, show_length):
+def handle_input(input_type, text, pdf, max_length, show_length):
     """
-    Handles the PDF upload, extracts text, and summarizes it.
+    Handles the user input based on the selected input type.
     
     Args:
+        input_type (str): The type of input (text or PDF).
+        text (str): The text input.
         pdf (UploadedFile): The uploaded PDF file.
         max_length (int): The maximum length of the summary.
         show_length (bool): Whether to show the length of the summary.
@@ -88,19 +90,24 @@ def handle_pdf(pdf, max_length, show_length):
     Returns:
         str: The summarized text.
     """
-    text = extract_text_from_pdf(pdf.name)
-    return summarize_text(text, max_length, show_length)
+    if input_type == 'Text':
+        return summarize_text(text, max_length, show_length)
+    elif input_type == 'PDF':
+        extracted_text = extract_text_from_pdf(pdf.name)
+        return summarize_text(extracted_text, max_length, show_length)
 
 # Define the Gradio interface
 interface = gr.Interface(
-    fn=handle_pdf,
+    fn=handle_input,
     inputs=[
-        gr.File(label='Upload PDF', type='file'),
+        gr.Radio(['Text', 'PDF'], label='Input Type', type='value'),
+        gr.Textbox(lines=10, placeholder='Enter Text Here...', label='Input Text', visible=True),
+        gr.File(label='Upload PDF', type='filepath', visible=True),
         gr.Slider(minimum=10, maximum=150, step=1, label='Max Length'),
         gr.Checkbox(label='Show summary length', value=False)
     ],
     outputs=gr.Textbox(label='Summarized Text'),
-    title='PDF Text Summarizer using T5-finetuned-dialogue_sumxx'
+    title='Text or PDF Summarizer using T5-finetuned-dialogue_sumxx'
 )
 
 # Launch the Gradio interface