Upload app.py

app.py

@@ -24,20 +24,26 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
 qa_model = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 # ✅ Upload and process PDF
-def upload_pdf(file):
+def upload_pdf(files):  # <== make sure it's 'files', not 'file'
     try:
-        filename = file.name
-        token = str(uuid.uuid4())
+        messages = []
 
-        pdf_reader = PyPDF2.PdfReader(file)
-        text = "".join([page.extract_text() or "" for page in pdf_reader.pages])
-        chunks = [text[i:i+500] for i in range(0, len(text), 500)]
+        for file in files:
+            filename = file.name
+            token = str(uuid.uuid4())
 
-        create_db(token, chunks, filename, text)
-        return f"✅ Uploaded and stored: {filename} (Token: {token})"
+            pdf_reader = PyPDF2.PdfReader(file)
+            text = "".join([page.extract_text() or "" for page in pdf_reader.pages])
+            chunks = [text[i:i+500] for i in range(0, len(text), 500)]
+
+            create_db(token, chunks, filename, text)
+            messages.append(f"✅ Uploaded: {filename} (Token: {token})")
+
+        return "\n".join(messages)
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
 # ✅ Generate Q&A using filename
 def generate_qa(filename):
     try:
@@ -123,13 +129,12 @@ with gr.Blocks(theme="default") as demo:
         """
     )
 
-    with gr.Tab("📤 1. Upload PDF"):
-        gr.Markdown("### 🗂 Upload a PDF File")
-        file = gr.File(label="Upload one or more PDFs", file_types=[".pdf"], file_count="multiple")
-
+    with gr.Tab("📁 1. Upload PDFs"):
+        file_input = gr.File(label="📂 Upload one or more PDFs", file_types=[".pdf"], file_count="multiple")
+        upload_output = gr.Markdown()
+        upload_btn = gr.Button("🚀 Upload")
 
-        upload_out = gr.Textbox(label="Upload Result", interactive=False)
-        file.change(fn=upload_pdf, inputs=file, outputs=upload_out)
+        upload_btn.click(fn=upload_pdf, inputs=file_input, outputs=upload_output)
 
     with gr.Tab("🧠 2. Generate Questions & Answers"):
         gr.Markdown("### 🤖 Generate Q&A from your PDF")