Update app.py

app.py

@@ -8,94 +8,84 @@ CHOSEN_BREED = "Golden Retrievers"
 # Page setup
 st.set_page_config(page_title=f"The {CHOSEN_BREED} Expert", page_icon="🐕")
 st.title(f"🐕 Only {CHOSEN_BREED}")
-st.caption(f"Powered by Perplexity API. I strictly refuse to talk about cats, other dogs, or anything else!")
+st.caption(f"Powered by Perplexity API. Strictly restricted to {CHOSEN_BREED}.")
 
 # --- AUTHENTICATION ---
 perplexity_api_key = os.getenv("PERPLEXITY_API_KEY")
 
 if not perplexity_api_key:
-    st.error("⚠️ PERPLEXITY_API_KEY not found in secrets. Please add it to your Hugging Face Space settings.")
+    st.error("⚠️ PERPLEXITY_API_KEY not found in secrets.")
     st.stop()
 
-# Initialize Client
 client = OpenAI(
     api_key=perplexity_api_key,
     base_url="https://api.perplexity.ai"
 )
 
-# --- STRICT SYSTEM PROMPT ---
-system_instruction = f"""
-You are a highly specialized AI assistant with a STRICT filter.
-You must adhere to the following funnel of allowed topics:
-
-1. LEVEL 1: Is the topic about Pets? (If NO -> Refuse)
-2. LEVEL 2: Is the topic about Dogs? (If NO -> Refuse)
-3. LEVEL 3: Is the topic about {CHOSEN_BREED}? (If NO -> Refuse)
-
-Your Goal: Provide accurate, helpful information specifically about {CHOSEN_BREED}.
-
-RESTRICTIONS:
-- If the user asks about Cats, Hamsters, or Birds: REFUSE. Say "I only talk about dogs."
-- If the user asks about German Shepherds, Pugs, or Labradors: REFUSE. Say "I am specialized only in {CHOSEN_BREED}."
-- If the user asks about Math, Weather, or Politics: REFUSE. Say "I only discuss {CHOSEN_BREED}."
-
-Refusal Message Style: Be polite but firm. Do not apologize profusely, just state your limitation.
-"""
-
 # --- CHAT LOGIC ---
-
-# Initialize chat history
 if "messages" not in st.session_state:
-    # We keep the greeting in the UI state, but we will filter it out for the API later
     st.session_state["messages"] = [
-        {"role": "assistant", "content": f"Hello! I am your expert on {CHOSEN_BREED}. Ask me anything about them!"}
+        {"role": "assistant", "content": f"Hello! I only answer questions about {CHOSEN_BREED}. Try me!"}
     ]
 
-# Display chat messages
 for msg in st.session_state.messages:
     st.chat_message(msg["role"]).write(msg["content"])
 
-# Handle new user input
-if prompt := st.chat_input():
+if user_input := st.chat_input():
     
-    # --- ERROR FIX: CLEANUP STALE MESSAGES ---
-    # If the last message was from the user, it means the previous API call failed.
-    # We must remove it to prevent sending "User, User" sequence which breaks the API.
+    # 1. Clean up previous failed user messages to prevent "User, User" errors
     if st.session_state.messages and st.session_state.messages[-1]["role"] == "user":
         st.session_state.messages.pop()
 
-    # 1. Display current user message
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
+    # 2. Display User Message
+    st.session_state.messages.append({"role": "user", "content": user_input})
+    st.chat_message("user").write(user_input)
 
-    # 2. Prepare messages for API
-    # Start with System Prompt
-    api_messages = [{"role": "system", "content": system_instruction}]
+    # 3. CONSTRUCT THE STRICT PROMPT (The "Secret Sauce")
+    # We don't just send the user input. We wrap it in a strict logic gate.
+    # We ask Perplexity to output "REFUSAL_TRIGGERED" if it fails the test.
     
-    # Add history, but SKIP the very first artificial greeting to ensure System -> User flow
-    # Perplexity prefers the conversation to start with a User query, not an Assistant greeting.
-    history_to_send = st.session_state.messages[-10:] # Get last 10
+    strict_content = f"""
+    Task: Check if the following query is specifically about the dog breed "{CHOSEN_BREED}".
     
-    for msg in history_to_send:
-        # Only append if it's NOT that initial fake greeting
-        if msg["content"] != f"Hello! I am your expert on {CHOSEN_BREED}. Ask me anything about them!":
-            api_messages.append(msg)
+    User Query: "{user_input}"
+    
+    RULES:
+    1. If the query is about "{CHOSEN_BREED}", answer it helpfully.
+    2. If the query is about general dogs (but not this specific breed), cats, weather, math, sports, or ANY other topic:
+       YOU MUST REPLY WITH EXACTLY ONE WORD: "REFUSAL_TRIGGERED"
+    
+    Do not explain. Do not search for the weather. Do not answer the question if it violates Rule 2.
+    """
+
+    # 4. Prepare Messages
+    # Note: We do NOT send the full chat history for the "Strict Check" to prevent it from getting confused by context.
+    # We treat every query as a fresh strict check.
+    api_messages = [
+        {"role": "system", "content": "You are a strict validation machine. You only output answers for Golden Retrievers."},
+        {"role": "user", "content": strict_content}
+    ]
 
-    # 3. Call Perplexity API
+    # 5. Call API
     try:
         response = client.chat.completions.create(
             model="sonar", 
             messages=api_messages,
-            temperature=0.1,
+            temperature=0.0, # Zero temperature for maximum strictness
         )
         
-        bot_reply = response.choices[0].message.content
-        
-        # 4. Display Assistant Response
-        st.session_state.messages.append({"role": "assistant", "content": bot_reply})
-        st.chat_message("assistant").write(bot_reply)
+        bot_reply = response.choices[0].message.content.strip()
+
+        # 6. POST-PROCESSING GUARDRAIL
+        # If the model obeyed our "REFUSAL_TRIGGERED" command, we intercept it.
+        if "REFUSAL_TRIGGERED" in bot_reply or "REFUSAL" in bot_reply:
+             final_reply = f"🚫 I cannot answer that. I am strictly programmed to discuss **{CHOSEN_BREED}** only."
+        else:
+             final_reply = bot_reply
+
+        # 7. Show Result
+        st.session_state.messages.append({"role": "assistant", "content": final_reply})
+        st.chat_message("assistant").write(final_reply)
 
     except Exception as e:
-        st.error(f"An error occurred: {e}")
-        # Optional: Print the message structure for debugging if needed
-        # st.write(api_messages)
+        st.error(f"An error occurred: {e}")