Update app.py

app.py

@@ -16,13 +16,13 @@ st.set_page_config(page_title="🐶 DUBSChat", layout="centered")
 with st.sidebar:
     if "dubs_key" not in st.session_state:
         st.session_state["dubs_key"] = ""
-    st.session_state["dubs_key"] = st.text_input("Enter Dubs Key", key="chatbot_api_key", type="password")
+    st.session_state["dubs_key"] = st.text_input("Enter Dubs Key", type="password", key="chatbot_api_key")
 
     st.markdown("### Chat Sessions")
     if "sessions" not in st.session_state:
         st.session_state["sessions"] = {"Default": ConversationBufferMemory()}
     session_keys = list(st.session_state["sessions"].keys())
-    selected_session = st.selectbox("Select Chat Session", session_keys)
+    selected_session = st.selectbox("Select Chat Session", session_keys, index=session_keys.index(st.experimental_get_query_params().get("session", ["Default"])[0]))
     new_session_name = st.text_input("New Session Name")
     if st.button("Create New Session"):
         if new_session_name and new_session_name not in st.session_state["sessions"]:
@@ -41,7 +41,7 @@ class CustomLLM(LLM):
     """
 
     def _call(self, prompt: str, stop: Optional[list] = None) -> str:
-        dubs_key = st.session_state.get("dubs_key", None)
+        dubs_key = st.session_state.get("dubs_key")
         if not dubs_key:
             return "Error: Missing Dubs Key. Please provide a valid key in the sidebar."
 
@@ -52,20 +52,9 @@ class CustomLLM(LLM):
         payload = {"prompt": prompt}
 
         try:
-            response = requests.post(SPACE_URL, json=payload, headers=headers, stream=True)
+            response = requests.post(SPACE_URL, json=payload, headers=headers)
             response.raise_for_status()
-
-            # Debugging: Log the raw response
-            raw_response = response.text
-            print("Raw Response:", raw_response)
-
-            # Handle non-JSON responses
-            try:
-                result = response.json()
-            except json.JSONDecodeError:
-                return f"Error decoding response: {raw_response}"
-
-            # Extract and return the API's response content
+            result = response.json()
             return result.get("response", "No response from the model.")
 
         except requests.exceptions.HTTPError as e: