Upload 7 files

app.py

@@ -24,29 +24,20 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
 qa_model = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 # ✅ Upload and process PDF
-
-# ✅ Multiple file upload logic
-def upload_pdf(files):
+def upload_pdf(file):
     try:
-        messages = []
-
-        for file in files:
-            filename = file.name
-            token = str(uuid.uuid4())
-
-            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 and stored: {filename} (Token: {token})")
+        filename = file.name
+        token = str(uuid.uuid4())
 
-        return "\n".join(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)]
 
+        create_db(token, chunks, filename, text)
+        return f"✅ Uploaded and stored: {filename} (Token: {token})"
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-
 # ✅ Generate Q&A using filename
 def generate_qa(filename):
     try:
@@ -77,7 +68,7 @@ def generate_qa(filename):
 # ✅ Ask question using token (semantic similarity)
 def ask_question(token, question):
     try:
-        with sqlite3.connect("my_database.db") as conn:
+        with sqlite3.connect("my_database1.db") as conn:
             cursor = conn.cursor()
             cursor.execute("SELECT chunk_data FROM token_data WHERE token_id = ?", (token,))
             row = cursor.fetchone()
@@ -102,20 +93,21 @@ 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:
+        with sqlite3.connect("my_database1.db") as conn:
             cursor = conn.cursor()
-            cursor.execute("SELECT filename FROM token_data")
+            cursor.execute("SELECT filename, token_id, content FROM token_data")
             rows = cursor.fetchall()
 
         if not rows:
             return "ℹ️ No PDFs uploaded yet."
 
-        result = "📂 **Uploaded PDF Files:**\n\n"
-        for i, (filename,) in enumerate(rows, start=1):
-            result += f"{i}. {filename}\n"
-
+        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"
         return result
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -131,25 +123,13 @@ with gr.Blocks(theme="default") as demo:
         """
     )
 
-    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("📤 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.Tab("🧠 2. Generate Questions & Answers"):
         gr.Markdown("### 🤖 Generate Q&A from your PDF")
@@ -165,17 +145,10 @@ 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")
-
+        gr.Markdown("### 📋 Uploaded PDF List (with Preview Snippets)")
         list_btn = gr.Button("🔄 Refresh List")
-
         pdf_list_output = gr.Markdown()
-
-        list_btn.click(
-            fn=list_uploaded_pdfs,
-            inputs=[],
-            outputs=pdf_list_output
-        )
+        list_btn.click(fn=list_uploaded_pdfs, outputs=pdf_list_output)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)

my_database1.db

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6412623c4a1e5bab6d73d3a7ac1f0deef76bb249d8e5b094f3095349d4a925e1
+size 6090752