Upload app.py

app.py

@@ -51,28 +51,20 @@ def upload_pdf(files):
 
 
 
-# ✅ 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 []
+# Load QG and QA once
+qgen = QGenerator()
+qa_model = pipeline("text2text-generation", model="google/flan-t5-base")
+
 
-# ✅ Main function to generate Q&A pairs from selected filename
 def generate_qa(filename):
     try:
         if not filename:
-            return "⚠️ Filename not provided."
+            return "⚠️ Please select a 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 LOWER(filename) = LOWER(?)", (filename,))
+            cursor.execute("SELECT chunk_data FROM token_data WHERE filename = ?", (filename,))
             row = cursor.fetchone()
 
         if not row:
@@ -86,11 +78,16 @@ def generate_qa(filename):
             if not questions:
                 continue
 
-            for question in questions[:2]:  # Max 2 questions per chunk
+            for question in questions[:2]:  # Max 2 Qs per chunk
                 prompt = f"Context: {chunk}\n\nQuestion: {question}\n\nAnswer:"
                 try:
                     result = qa_model(prompt, max_length=256, do_sample=False)
-                    answer = result[0]["generated_text"].strip()
+                    if isinstance(result, list) and "generated_text" in result[0]:
+                        answer = result[0]["generated_text"].strip()
+                    elif isinstance(result, dict) and "answer" in result:
+                        answer = result["answer"].strip()
+                    else:
+                        answer = "N/A"
                     qa_pairs.append(f"Q: {question}\nA: {answer}")
                 except Exception as e:
                     print(f"QA model failed: {e}")
@@ -104,7 +101,6 @@ def generate_qa(filename):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-
 # ✅ Ask question using token (semantic similarity)
 def ask_question(token, question):
     try:
@@ -177,16 +173,12 @@ 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.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.Blocks(title="PDF Q&A Generator") as demo:
+        with gr.Tab("🧠 2. Generate Questions & Answers"):
+            gr.Markdown("### 🤖 Generate Questions and Answers from Uploaded PDF")
+            fname = gr.Textbox(label="📄 Enter Uploaded Filename", placeholder="example.pdf")
+            output_box = gr.Textbox(label="📝 Generated Q&A", lines=15, interactive=False)
+            gr.Button("🚀 Generate Q&A").click(fn=generate_qa, inputs=fname, outputs=output_box)
 
     with gr.Tab("❓ 3. Ask a Question"):
         gr.Markdown("### 💬 Ask a question based on uploaded PDF")