Final: Updated app.py

app.py

@@ -151,9 +151,22 @@ def ask_question(question):
 
     context = "\n".join([doc.page_content for doc in docs])
     prompt = generate_prompt(context, question)
-    result = reasoning_pipeline(prompt)[0]['generated_text']
+    raw_output = reasoning_pipeline(prompt)[0]['generated_text']
+
+    # 🧠 Smart cleanup: remove prompt leakage or echoed instructions
+    for section in ["Context:", "Question:", "Instructions:", "Use structured academic language"]:
+        raw_output = raw_output.replace(section, "").strip()
+
+    # Remove anything before answer starts if needed
+    if "Answer:" in raw_output:
+        raw_output = raw_output.split("Answer:")[-1].strip()
+
+    # Trim trailing junk
+    if "." in raw_output:
+        raw_output = raw_output.rsplit(".", 1)[0] + "."
+
+    return post_process_output(raw_output.strip(), question)
 
-    return post_process_output(result, question)
 
 # Gradio UI
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo: