Upload 7 files

app.py

@@ -24,7 +24,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
 qa_model = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 # ✅ Upload and process PDF
-# ✅ Updated version – supports multiple PDF files
+
+# ✅ Multiple file upload logic
 def upload_pdf(files):
     try:
         messages = []
@@ -101,21 +102,20 @@ def ask_question(token, question):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# ✅ View uploaded PDFs (Gradio Tab)
 def list_uploaded_pdfs():
     try:
         with sqlite3.connect("my_database.db") as conn:
             cursor = conn.cursor()
-            cursor.execute("SELECT filename, token_id, content FROM token_data")
+            cursor.execute("SELECT filename FROM token_data")
             rows = cursor.fetchall()
 
         if not rows:
             return "ℹ️ No PDFs uploaded yet."
 
-        result = ""
-        for filename, token, content in rows:
-            preview = content[:200].replace("\n", " ") + "..." if len(content) > 200 else content
-            result += f"📄 **Filename:** {filename}\n🔑 Token: `{token}`\n📝 Preview: {preview}\n\n---\n"
+        result = "📂 **Uploaded PDF Files:**\n\n"
+        for i, (filename,) in enumerate(rows, start=1):
+            result += f"{i}. {filename}\n"
+
         return result
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -131,12 +131,25 @@ 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")
-       
-        upload_out = gr.Textbox(label="Upload Result", interactive=False)
-        file.change(fn=upload_pdf, inputs=file, outputs=upload_out)
+    with gr.Blocks(title="PDF Uploader") as demo:
+        with gr.Tab("📤 Upload PDFs"):
+            gr.Markdown("### Select and upload one or more PDF files")
+
+            file_input = gr.File(
+                label="Upload PDFs",
+                file_types=[".pdf"],
+                file_count="multiple"  # ✅ enables multi-select in UI
+            )
+
+            upload_output = gr.Textbox(
+                label="Upload Results",
+                lines=10,
+                interactive=False
+            )
+
+            file_input.change(fn=upload_pdf, inputs=file_input, outputs=upload_output)
+
+
 
     with gr.Tab("🧠 2. Generate Questions & Answers"):
         gr.Markdown("### 🤖 Generate Q&A from your PDF")
@@ -152,10 +165,17 @@ with gr.Blocks(theme="default") as demo:
         gr.Button("🎯 Get Answer").click(fn=ask_question, inputs=[token_box, question_box], outputs=answer_result)
 
     with gr.Tab("📚 4. View Uploaded PDFs"):
-        gr.Markdown("### 📋 Uploaded PDF List (with Preview Snippets)")
+        gr.Markdown("### 📋 Uploaded PDF List")
+
         list_btn = gr.Button("🔄 Refresh List")
+
         pdf_list_output = gr.Markdown()
-        list_btn.click(fn=list_uploaded_pdfs, outputs=pdf_list_output)
+
+        list_btn.click(
+            fn=list_uploaded_pdfs,
+            inputs=[],
+            outputs=pdf_list_output
+        )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)