Update app.py

app.py

@@ -84,6 +84,21 @@ def ensure_english(text):
         return "⚠️ Language detection failed. Please ask your question again."
     return text
 
+# ✅ Ensure Every Response Has a Follow-Up Question
+def generate_follow_up(user_text):
+    """Generates a follow-up question to guide the user toward related topics or next steps."""
+    prompt_text = (
+        f"Given the user's question: '{user_text}', generate a SHORT follow-up question "
+        "suggesting either a related topic or asking if they need further help. "
+        "Example: 'Would you like to explore quantum superposition or ask about another physics concept?' "
+        "Keep it concise and engaging."
+    )
+    hf = get_llm_hf_inference(max_new_tokens=40, temperature=0.8)
+    output = hf.invoke(input=prompt_text).strip()
+    
+    # Fallback in case of an empty response
+    return output if output else "Would you like to explore another related topic or ask about something else?"
+
 # ✅ Main Response Function
 def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     action = predict_action(user_text)  # 🔥 Fix: Define 'action'