Update app.py

app.py

@@ -6,7 +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")
 
@@ -14,10 +13,7 @@ st.set_page_config(page_title="💬 DUBSChat", layout="centered")
 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!""")
-
-
+    st.markdown("""Coming Soon!""")
 
 def call_api(message):
     """
@@ -26,8 +22,8 @@ def call_api(message):
     Args:
         message (str): User's message.
     
-    Returns:
-        str: Streamed response from the backend API or an error message.
+    Yields:
+        str: Streamed response chunks from the backend API.
     """
     headers = {
         "Authorization": f"Bearer {dubs_key}",
@@ -40,8 +36,6 @@ def call_api(message):
         response = requests.post(SPACE_URL, json=payload, headers=headers, stream=True)
         response.raise_for_status()
 
-        # Aggregate streamed response
-        bot_response = ""
         for chunk in response.iter_lines(decode_unicode=True):
             if chunk.strip():  # Ensure the chunk is not empty
                 try:
@@ -50,13 +44,11 @@ def call_api(message):
                     if "response" in data:
                         yield data["response"]
                 except json.JSONDecodeError:
-                    bot_response += "\nError decoding response chunk."
-        
-        return bot_response.strip() if bot_response else "Error: No response from API."
+                    yield "Error decoding response chunk."
     except requests.exceptions.Timeout:
-        return "Error: The API call timed out."
+        yield "Error: The API call timed out."
     except requests.exceptions.RequestException as e:
-        return f"Error: {str(e)}"
+        yield f"Error: {str(e)}"
 
 # Chat History Initialization
 if "messages" not in st.session_state:
@@ -81,9 +73,12 @@ if prompt := st.chat_input():
 
     with st.spinner("Translating Dubs language (Woof Woof!)"):
         assistant_message_placeholder = st.chat_message("assistant").empty()
-        response_stream = call_api(prompt)
-        assistant_message_placeholder.write_stream(response_stream)
+        full_response = ""  # To accumulate the entire response
+        for response_chunk in call_api(prompt):
+            full_response += response_chunk
+            assistant_message_placeholder.write(full_response)  # Update progressively
+
+        # Append the final assistant's response to the chat history
+        st.session_state["messages"].append({"role": "assistant", "content": full_response})
 
-    # Append the final assistant's response to the chat history
-    st.session_state["messages"].append({"role": "assistant", "content": response_stream})