Update app.py

app.py

@@ -10,18 +10,13 @@ DUBS_PATH = "🐾"  # Optional: Replace with an avatar path if needed
 # Streamlit Configuration
 st.set_page_config(page_title="Chatbot Test", page_icon="🤖", layout="centered")
 
-# Chat State
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
-
-client = InferenceClient(SPACE_URL, token=HF_API_KEY)
-
 # Function to Fetch Response (Non-Streaming)
 def fetch_response(prompt_text, api_key):
     """
     Fetch full text response from the HF Inference Endpoint using the InferenceClient.
     Returns the generated text as a whole.
     """
+    client = InferenceClient(SPACE_URL, token=api_key)
 
     gen_kwargs = {
         "max_new_tokens": 512,
@@ -45,20 +40,14 @@ st.title("Chatbot Testing Interface")
 prompt = st.chat_input("Enter your message...")
 
 if prompt:
-    # 1) Add the user's message to session state
-    st.session_state["messages"].append({"role": "user", "content": prompt})
+    # Display the user's message
     st.chat_message("user").write(prompt)
 
-    # 2) Build combined chat history for the model prompt
-    chat_history = "".join(
-        [f"<|{msg['role']}|>{msg['content']}<|end|>" for msg in st.session_state["messages"]]
-    )
+    # Build the chat history (use prompt directly for stateless behavior)
+    chat_history = f"<|user|>{prompt}<|end|>"
 
-    # 3) Generate the response
+    # Generate the response
     with st.chat_message("assistant", avatar=DUBS_PATH):
         with st.spinner("Dubs is thinking... Woof Woof! 🐾"):
             full_response = fetch_response(chat_history, HF_API_KEY)
             st.write(full_response)
-
-    # 4) Add assistant response to the session state
-    st.session_state["messages"].append({"role": "assistant", "content": full_response})