Upload 6 files

app.py

@@ -13,7 +13,7 @@ from database1 import create_db
 from first1 import pdf_query
 from q_generator1 import QGenerator
 from ans_generator1 import AnswerGenerator
-from flask import  request, jsonify
+from flask import Flask, request, jsonify
 import sqlite3, json
 from q_generator1 import QGenerator
 from transformers import pipeline
@@ -56,12 +56,10 @@ qgen = QGenerator()
 qa_model = pipeline("text2text-generation", model="google/flan-t5-base")
 
 
-def generate_qa():
+def generate_qa(filename):
     try:
-        filename = request.form.get("filename")
-
         if not filename:
-            return jsonify({"error": "⚠️ Filename not provided."}), 400
+            return "⚠️ Filename not provided."
 
         # Load chunk_data from DB
         with sqlite3.connect("my_database.db") as conn:
@@ -70,7 +68,7 @@ def generate_qa():
             row = cursor.fetchone()
 
         if not row:
-            return jsonify({"error": "❌ No data found for this filename."}), 404
+            return "❌ No data found for this filename."
 
         chunks = json.loads(row[0])
         qa_pairs = []
@@ -85,24 +83,19 @@ def generate_qa():
                 try:
                     result = qa_model(prompt, max_length=256, do_sample=False)
                     answer = result[0]["generated_text"].strip()
-                    qa_pairs.append({
-                        "question": question,
-                        "answer": answer
-                    })
+                    qa_pairs.append(f"Q: {question}\nA: {answer}")
                 except Exception as e:
                     print(f"QA model failed: {e}")
                     continue
 
         if not qa_pairs:
-            return jsonify({"message": "⚠️ No Q&A pairs generated."}), 200
+            return "⚠️ No Q&A pairs generated."
 
-        return jsonify({
-            "filename": filename,
-            "qa_pairs": qa_pairs
-        }), 200
+        return "\n\n".join(qa_pairs)
 
     except Exception as e:
-        return jsonify({"error": f"❌ Error: {str(e)}"}), 500
+        return f"❌ Error: {str(e)}"
+
 
 
 # ✅ Ask question using token (semantic similarity)