Update app.py

app.py

@@ -2,20 +2,31 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# Load HF token from environment
 HF_TOKEN = os.getenv("HF_TOKEN")
 
+# Allowed subjects keywords + common Chemistry, Physics, Maths, Biology terms
+ALLOWED_KEYWORDS = [
+    "biology", "physics", "chemistry", "math", "mathematics",
+    "acid", "base", "neutralization", "molecule", "atom", "cell",
+    "force", "energy", "gravity", "equation", "algebra", "calculus", "integral", "derivative"
+]
+
 def respond(message, history: list[dict[str, str]]):
+    # Check if message contains allowed topics/keywords
+    if not any(keyword in message.lower() for keyword in ALLOWED_KEYWORDS):
+        yield "❌ Sorry, I only know about Biology, Physics, Chemistry, and Maths. Please ask me about them."
+        return
+
+    # Allowed topic → send to GPT model
     client = InferenceClient(token=HF_TOKEN, model="openai/gpt-oss-20b")
-    
-    # Prepare messages for chat
-    messages = [{"role": "system", "content": "You are a friendly chatbot."}]
+
+    messages = [
+        {"role": "system", "content": "You are a friendly chatbot. You only answer questions about Biology, Physics, Chemistry, and Maths. For anything else, politely reply that you can't answer it."}
+    ]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
-    
+
     response = ""
-    
-    # Stream response
     for msg in client.chat_completion(
         messages,
         max_tokens=512,
@@ -25,7 +36,7 @@ def respond(message, history: list[dict[str, str]]):
     ):
         if msg.choices and msg.choices[0].delta.content:
             response += msg.choices[0].delta.content
-            yield response
+        yield response
 
 # 💜 Modern CSS + Footer Hide + Compact Layout
 custom_css = """
@@ -96,16 +107,15 @@ button:hover {
     to {opacity: 1; transform: translateY(0);}
 }
 
-/* 🔒 Hide Gradio Footer (API, Logo, Settings) only */
+/* 🔒 Hide Gradio Footer */
 footer, .svelte-1yycg3h, .builder-bar, .wrap.svelte-1ipelgc {
     display: none !important;
     visibility: hidden !important;
 }
 """
 
-# Launch Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
-    gr.ChatInterface(respond, type="messages")  # Only chat interface, no header
+    gr.ChatInterface(respond, type="messages")
 
 if __name__ == "__main__":
     demo.launch(share=True)