Upload app.py

app.py

@@ -51,11 +51,19 @@ def upload_pdf(files):
 
 
 
-# Load QG and QA once
-qgen = QGenerator()
-qa_model = pipeline("text2text-generation", model="google/flan-t5-base")
-
+# ✅ Helper function to get uploaded filenames from DB
+def get_filenames():
+    try:
+        with sqlite3.connect("my_database.db") as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT DISTINCT filename FROM token_data")
+            rows = cursor.fetchall()
+            return [row[0] for row in rows] if rows else []
+    except Exception as e:
+        print("Error fetching filenames:", e)
+        return []
 
+# ✅ Main function to generate Q&A pairs from selected filename
 def generate_qa(filename):
     try:
         if not filename:
@@ -64,7 +72,7 @@ def generate_qa(filename):
         # Load chunk_data from DB
         with sqlite3.connect("my_database.db") as conn:
             cursor = conn.cursor()
-            cursor.execute("SELECT chunk_data FROM token_data WHERE filename = ?", (filename,))
+            cursor.execute("SELECT chunk_data FROM token_data WHERE LOWER(filename) = LOWER(?)", (filename,))
             row = cursor.fetchone()
 
         if not row:
@@ -78,7 +86,7 @@ def generate_qa(filename):
             if not questions:
                 continue
 
-            for question in questions[:2]:  # Max 2 Qs per chunk
+            for question in questions[:2]:  # Max 2 questions per chunk
                 prompt = f"Context: {chunk}\n\nQuestion: {question}\n\nAnswer:"
                 try:
                     result = qa_model(prompt, max_length=256, do_sample=False)
@@ -97,7 +105,6 @@ def generate_qa(filename):
         return f"❌ Error: {str(e)}"
 
 
-
 # ✅ Ask question using token (semantic similarity)
 def ask_question(token, question):
     try:
@@ -151,6 +158,7 @@ def list_uploaded_pdfs():
         return f"❌ Error: {str(e)}"
 
 
+
 # ✅ Gradio UI
 with gr.Blocks(theme="default") as demo:
     gr.Markdown(
@@ -169,11 +177,16 @@ with gr.Blocks(theme="default") as demo:
         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")
-        fname = gr.Textbox(label="Enter uploaded filename", placeholder="example.pdf")
-        qa_result = gr.Textbox(label="Generated Q&A", lines=15, interactive=False)
-        gr.Button("🚀 Generate Q&A").click(fn=generate_qa, inputs=fname, outputs=qa_result)
+
+    
+        with gr.Blocks(title="PDF Q&A Generator") as demo:
+            with gr.Tab("🧠 2. Generate Questions & Answers"):
+                gr.Markdown("### 🤖 Generate Q&A from your uploaded PDF")
+
+                fname = gr.Dropdown(label="Choose uploaded PDF", choices=get_filenames)
+                qa_result = gr.Textbox(label="Generated Q&A", lines=15, interactive=False)
+                gr.Button("🚀 Generate Q&A").click(fn=generate_qa, inputs=fname, outputs=qa_result)
+
 
     with gr.Tab("❓ 3. Ask a Question"):
         gr.Markdown("### 💬 Ask a question based on uploaded PDF")