Add text input and improved UI

app.py

@@ -7,8 +7,9 @@ import tempfile
 import os
 
 # Page config
-st.set_page_config(page_title="PDF & Image Summarizer", layout="centered")
-st.title("📄🖼 Document Summarizer")
+st.set_page_config(page_title="Docurizzer - Document Summarizer", layout="centered")
+st.title("📄🖼 Docurizzer")
+st.markdown("*Extract and summarize text from documents, images, or paste your own text*")
 
 # Load summarization model
 @st.cache_resource
@@ -17,12 +18,6 @@ def load_model():
 
 summarizer = load_model()
 
-# File uploader
-uploaded_file = st.file_uploader(
-    "Upload PDF or Image",
-    type=["pdf", "png", "jpg", "jpeg"]
-)
-
 def extract_text_from_image(image):
     return pytesseract.image_to_string(image)
 
@@ -35,31 +30,82 @@ def extract_text_from_pdf(file_path):
                 text += page_text
     return text
 
-if uploaded_file:
-    # Save temp file
-    with tempfile.NamedTemporaryFile(delete=False) as tmp:
-        tmp.write(uploaded_file.read())
-        file_path = tmp.name
+# Input method selection
+input_method = st.radio(
+    "Choose input method:",
+    ["📝 Paste Text", "📄 Upload PDF", "🖼 Upload Image"],
+    horizontal=True
+)
+
+text = ""
 
-    ext = os.path.splitext(uploaded_file.name)[1].lower()
+if input_method == "📝 Paste Text":
+    st.subheader("📝 Enter Your Text")
+    text = st.text_area(
+        "Paste or type your text here:",
+        height=250,
+        placeholder="Enter the text you want to summarize..."
+    )
 
-    if ext == ".pdf":
+elif input_method == "📄 Upload PDF":
+    uploaded_file = st.file_uploader(
+        "Upload a PDF file",
+        type=["pdf"]
+    )
+    
+    if uploaded_file:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+            tmp.write(uploaded_file.read())
+            file_path = tmp.name
+        
         text = extract_text_from_pdf(file_path)
-    else:
-        image = Image.open(file_path)
-        text = extract_text_from_image(image)
+        
+        st.subheader("📃 Extracted Text (Preview)")
+        st.text_area("", text[:3000], height=200, disabled=True)
 
-    st.subheader("📃 Extracted Text (Preview)")
-    st.text_area("", text[:3000], height=200)
+elif input_method == "🖼 Upload Image":
+    uploaded_file = st.file_uploader(
+        "Upload an image (PNG, JPG, JPEG)",
+        type=["png", "jpg", "jpeg"]
+    )
+    
+    if uploaded_file:
+        with tempfile.NamedTemporaryFile(delete=False) as tmp:
+            tmp.write(uploaded_file.read())
+            file_path = tmp.name
+        
+        image = Image.open(file_path)
+        st.image(image, caption="Uploaded Image", use_container_width=True)
+        
+        with st.spinner("Extracting text from image..."):
+            text = extract_text_from_image(image)
+        
+        st.subheader("📃 Extracted Text (Preview)")
+        st.text_area("", text[:3000], height=200, disabled=True)
 
-    if st.button("Summarize"):
+# Summarize button
+if text.strip():
+    if st.button("🚀 Summarize", type="primary"):
         with st.spinner("Summarizing..."):
+            # Handle text length for T5 model
+            input_text = text[:4000] if len(text) > 4000 else text
+            
             summary = summarizer(
-                text,
-                max_length=120,
+                input_text,
+                max_length=150,
                 min_length=40,
                 do_sample=False
             )[0]["summary_text"]
 
         st.subheader("📝 Summary")
         st.success(summary)
+        
+        # Copy button
+        st.download_button(
+            label="📋 Download Summary",
+            data=summary,
+            file_name="summary.txt",
+            mime="text/plain"
+        )
+else:
+    st.info("👆 Please provide some text using one of the methods above.")