Update app.py

app.py

@@ -12,13 +12,24 @@ def extract_text_from_pdf(pdf_content):
     return text
 
 
-def generate_summary(file_content):
-    text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
-    if file_content.type == "application/pdf":
-        input_text = extract_text_from_pdf(file_content.read())
+def generate_summary(file_content, user_input_text):
+    if file_content:
+        # Handle uploaded file
+        if isinstance(file_content, bytes):
+            input_text = extract_text_from_pdf(file_content)
+        else:
+            input_text = file_content.read().decode("utf-8")
     else:
-        input_text = file_content.read().decode("utf-8")
+        # Handle user input text
+        input_text = user_input_text
+
+    # Load pre-trained Flan-T5 model
+    text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
+
+    # Generate summary
     summary = text_generator(input_text, max_length=1024, num_beams=4)
+
+    # Return results
     return {
         "Extracted Information": input_text,
         "Book Summary": summary[0]["generated_text"],
@@ -28,7 +39,7 @@ def generate_summary(file_content):
 
 iface = gr.Interface(
     fn=generate_summary,
-    inputs=gr.File(),
+    inputs=[gr.File(label="Upload File"), gr.Textbox(label="Enter Text")],
     outputs=[
         gr.Textbox(),
         gr.Textbox(),