updated question structure

database.py

@@ -55,8 +55,10 @@ def answer_question(context, question):
     """
     prompt = f"{context}\n\nUser Question: {question}\nSQL Query or Answer:"
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
-    outputs = model.generate(inputs.input_ids, max_length=512, num_beams=5, early_stopping=True)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    outputs = model.generate(inputs.input_ids, max_length=1024, num_beams=5, early_stopping=True)
+    answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    return answer
 
 # Interactive loop
 print("Database Assistant is ready. Ask your questions!")