Update app.py

app.py

@@ -1,3 +1,21 @@
+import streamlit as st
+from huggingface_hub import InferenceClient
+import os
+
+# Constants
+SPACE_URL = "https://z7svds7k42bwhhgm.us-east-1.aws.endpoints.huggingface.cloud"
+HF_API_KEY = os.getenv("HF_API_KEY")
+DUBS_PATH = "🐾"  # Optional: Replace with an avatar path if needed
+
+# Streamlit Configuration
+st.set_page_config(page_title="Chatbot Test", page_icon="🤖", layout="centered")
+
+client = InferenceClient(SPACE_URL, token=HF_API_KEY)
+
+
+
+    
+
 def fetch_response(prompt):
     """
     Fetch full text response from the HF Inference Endpoint using the InferenceClient.
@@ -23,3 +41,35 @@ def fetch_response(prompt):
             break
         partial_text += response.token.text
         yield response.token.text
+
+
+
+
+
+# Streamlit Chat Interface
+st.title("Chatbot Testing Interface")
+
+# User Input Section
+prompt = st.chat_input("Enter your message...")
+
+if prompt:
+    # Display the user's message
+    st.chat_message("user").write(prompt)
+
+    # Build the chat history (use prompt directly for stateless behavior)
+    chat_history = f"<|user|>{prompt}<|end|> \n <|assistant|> "
+
+    # Generate the response
+    with st.spinner("Dubs is thinking... Woof Woof! 🐾"):
+        with st.chat_message("assistant", avatar=DUBS_PATH):
+            full_response = fetch_response(chat_history)
+            st.write_stream(full_response)
+
+
+
+
+
+
+
+
+