Update app.py

app.py

@@ -1,5 +1,51 @@
 import streamlit as st
-from api_helper import call_api  # Keep the original API logic
+import requests
+import json
+
+# Constants
+SPACE_URL = "https://abanm-dubs.hf.space/api/generate"
+API_KEY = "s3cr3t_k3y"  # Replace with your actual API key
+
+def call_api(message):
+    """
+    Calls the backend API with the user's input using streaming.
+    
+    Args:
+        message (str): User's message.
+    
+    Returns:
+        str: Streamed response from the backend API or an error message.
+    """
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json",
+    }
+    payload = {"prompt": message}  # Match the API's expected input key
+
+    try:
+        # Send request with streaming enabled
+        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:
+                    # Parse JSON chunk
+                    data = json.loads(chunk)
+                    if "response" in data:
+                        bot_response += data["response"]
+                        yield bot_response.strip()
+                except json.JSONDecodeError:
+                    bot_response += "\nError decoding response chunk."
+        
+        if not bot_response:
+            yield "Error: No response from API."
+    except requests.exceptions.Timeout:
+        yield "Error: The API call timed out."
+    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")
@@ -31,10 +77,12 @@ if prompt := st.chat_input():
     st.session_state["messages"].append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
-    with st.spinner("Translating to Dubs language(Woof Woof!)"):
-        # Call the original API logic to get the response
-        response = call_api(prompt)
+    with st.spinner("Translating to Dubs language (Woof Woof!)"):
+        # Call the API logic to get the response with streaming
+        bot_response = ""
+        for partial_response in call_api(prompt):
+            bot_response = partial_response
+            st.chat_message("assistant").write(bot_response)
 
-    # Append the assistant's response to the chat history
-    st.session_state["messages"].append({"role": "assistant", "content": response})
-    st.chat_message("assistant").write(response)
+    # Append the final assistant's response to the chat history
+    st.session_state["messages"].append({"role": "assistant", "content": bot_response})