Update app.py

app.py

@@ -1,5 +1,47 @@
+import requests
+import json
 import streamlit as st
-from api_helper import call_api  # Import the updated API helper
+
+# 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.
+
+    Yields:
+        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()
+
+        # Stream each line of the 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"]
+                except json.JSONDecodeError:
+                    yield "\nError decoding response chunk."
+
+    except requests.exceptions.Timeout:
+        yield "Error: The API call timed out."
+    except requests.exceptions.RequestException as e:
+        yield f"Error: {str(e)}"
 
 # Streamlit Configurations
 st.set_page_config(page_title="Chatbot", layout="centered")
@@ -15,29 +57,27 @@ def handle_input():
         # Save User Message
         st.session_state["messages"].append({"role": "user", "content": user_message})
 
-
-        # Display Bot Response Placeholder
-        response_placeholder = st.empty()  # Create an empty container for dynamic updates
+        # Placeholder for dynamic bot response
+        response_placeholder = st.empty()  # Create a dynamic container
         bot_response = ""
 
         with st.spinner("Fetching response..."):
             for part in call_api(user_message):
-                bot_response += part  # Append each part to the complete response
+                bot_response += part  # Append each part to the full response
                 response_placeholder.markdown(
-                f"<div style='text-align: left; color: green; margin: 10px 0;'><b>Bot:</b> {bot_response}</div>",
-                unsafe_allow_html=True,
-        )
+                    f"<div style='text-align: left; color: green; margin: 10px 0;'><b>Bot:</b> {bot_response}</div>",
+                    unsafe_allow_html=True,
+                )
 
         # Save Final Bot Response
         st.session_state["messages"].append({"role": "assistant", "content": bot_response})
 
-
         # Clear Input Field
         st.session_state.user_input = ""  # Reset user input field
 
 # Main Chat UI
 st.title("DUBS v.0.0.3")
-st.markdown("Emnpowering you with a Sustainable AI ")
+st.markdown("Empowering you with a Sustainable AI")
 
 # Chat Display
 for message in st.session_state["messages"]:
@@ -59,3 +99,4 @@ st.text_input(
     key="user_input",
     on_change=handle_input,
 )
+