Update app.py

app.py

@@ -6,6 +6,19 @@ import json
 SPACE_URL = "https://abanm-dubs.hf.space/api/generate"
 API_KEY = "s3cr3t_k3y"  # Replace with your actual API key
 
+
+# Streamlit Configurations (must come first)
+st.set_page_config(page_title="💬 DUBSChat", layout="centered")
+
+# Sidebar Configuration
+with st.sidebar:
+    dubs_key = st.text_input("Enter Dubs Key", key="chatbot_api_key", type="password")
+    st.markdown("Dubs Recall")
+    #st.markdown("""Other Features.""")
+    st.markdown("""Comming Soon!""")
+
+
+
 def call_api(message):
     """
     Calls the backend API with the user's input using streaming.
@@ -13,11 +26,11 @@ def call_api(message):
     Args:
         message (str): User's message.
     
-    Yields:
-        str: Streamed partial response from the backend API.
+    Returns:
+        str: Streamed response from the backend API or an error message.
     """
     headers = {
-        "Authorization": f"Bearer {API_KEY}",
+        "Authorization": f"Bearer {dubs_key}",
         "Content-Type": "application/json",
     }
     payload = {"prompt": message}  # Match the API's expected input key
@@ -27,20 +40,23 @@ def call_api(message):
         response = requests.post(SPACE_URL, json=payload, headers=headers, stream=True)
         response.raise_for_status()
 
-        # Stream response chunks
+        # Aggregate streamed response
+        bot_response = ""
         for chunk in response.iter_lines(decode_unicode=True):
             if chunk.strip():  # Ensure the chunk is not empty
                 try:
                     # Parse JSON chunk
                     data = json.loads(chunk)
                     if "response" in data:
-                        yield data["response"]
+                        bot_response += data["response"]
                 except json.JSONDecodeError:
-                    yield "\nError decoding response chunk."
+                    bot_response += "\nError decoding response chunk."
+        
+        return bot_response.strip() if bot_response else "Error: No response from API."
     except requests.exceptions.Timeout:
-        yield "Error: The API call timed out."
+        return "Error: The API call timed out."
     except requests.exceptions.RequestException as e:
-        yield f"Error: {str(e)}"
+        return f"Error: {str(e)}"
 
 # Streamlit Configurations (must come first)
 st.set_page_config(page_title="💬 DUBSChat", layout="centered")
@@ -56,7 +72,7 @@ if "messages" not in st.session_state:
     st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
 
 # Main Chat UI
-st.title("DubsChat")
+st.title("DUBS v.0.0.3")
 st.markdown("Empowering you with a Sustainable AI")
 
 # Chat Display
@@ -75,17 +91,12 @@ if prompt := st.chat_input():
     # Display a placeholder for the assistant's response
     assistant_message_placeholder = st.chat_message("assistant").empty()
 
-    bot_response = ""
-    first_chunk = True
-
-    for partial_response in call_api(prompt):
-        bot_response += partial_response
-
-        if first_chunk:
-            first_chunk = False
-            st.spinner("Translating from Dubs language (Woof Woof!)")
-
-        assistant_message_placeholder.write(bot_response)  # Update incrementally
+    with st.spinner("Translating Dubs language (Woof Woof!)"):
+        bot_response = ""
+        for partial_response in call_api(prompt):
+            bot_response += partial_response
+            assistant_message_placeholder.write(bot_response)  # Update incrementally
 
     # Append the final assistant's response to the chat history
     st.session_state["messages"].append({"role": "assistant", "content": bot_response})
+