Update app.py

app.py

@@ -6,19 +6,6 @@ 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 your Dubs Key", key="chatbot_api_key", type="password")
-    #st.markdown("### DUBS Settings")
-    st.markdown("""Dubs Recall""")
-    st.markdown("""Comming Soon!""")
-
 def call_api(message):
     """
     Calls the backend API with the user's input using streaming.
@@ -30,7 +17,7 @@ def call_api(message):
         str: Streamed partial response from the backend API.
     """
     headers = {
-        "Authorization": f"Bearer {Dubs_Key}",
+        "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json",
     }
     payload = {"prompt": message}  # Match the API's expected input key
@@ -55,16 +42,21 @@ def call_api(message):
     except requests.exceptions.RequestException as e:
         yield f"Error: {str(e)}"
 
+# Streamlit Configurations (must come first)
+st.set_page_config(page_title="💬 DUBSChat", layout="centered")
 
-
-
+# Sidebar Configuration
+with st.sidebar:
+    st.markdown("### DUBS Settings")
+    st.markdown("""Other Features.""")
+    st.markdown("""Comming Soon!""")
 
 # Chat History Initialization
 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("DUBS v.0.0.3")
+st.title("DubsChat")
 st.markdown("Empowering you with a Sustainable AI")
 
 # Chat Display
@@ -83,11 +75,17 @@ if prompt := st.chat_input():
     # Display a placeholder for the assistant's response
     assistant_message_placeholder = st.chat_message("assistant").empty()
 
-    with st.spinner("Translating to 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
+    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
 
     # Append the final assistant's response to the chat history
     st.session_state["messages"].append({"role": "assistant", "content": bot_response})