deepkansara-123 commited on
Commit
48c487d
·
verified ·
1 Parent(s): 090906c

Update app.py

Browse files

question answer the in file name to extrect data

Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -90,15 +90,13 @@ def generate_qa(token):
90
  # ✅ Ask a question using token
91
  def ask_question(token, question):
92
  try:
93
- with sqlite3.connect("my_database.db") as conn:
94
- cursor = conn.cursor()
95
- cursor.execute("SELECT chunk_data FROM token_data WHERE token_id = ?", (token,))
96
- row = cursor.fetchone()
97
 
98
  if not row:
99
  return "❌ Token not found."
100
 
101
- chunks = json.loads(row[0])
102
  processor = pdf_query()
103
  model = processor.model
104
 
 
90
  # ✅ Ask a question using token
91
  def ask_question(token, question):
92
  try:
93
+
94
+ row=create_db.fetch_by_token_or_filename(token)
 
 
95
 
96
  if not row:
97
  return "❌ Token not found."
98
 
99
+ chunks = json.loads(row['chunk_data'])
100
  processor = pdf_query()
101
  model = processor.model
102