Bug fix

app.py

@@ -7,21 +7,26 @@ import urllib.parse
 
 st.title("Prem Chat UI")
 
-if "api_key" not in st.session_state and "project_id" not in st.session_state:
+if "api_key" not in st.session_state or "project_id" not in st.session_state:
     api_key = st.text_input("Enter your API Key", type="password")
     project_id = st.text_input("Enter your Project ID")
-    if api_key and project_id:
-        if project_id.isdecimal():
-            st.session_state.api_key = api_key
-            st.session_state.project_id = int(project_id)
-            client = Prem(api_key=api_key)
-            st.rerun()
-        else:
-            st.warning("Please enter your Project ID correctly.")
-            st.stop()
-    else:
+    
+    if not api_key or not project_id:
         st.warning("Please enter your API key and Project ID to use the app.")
         st.stop()
+
+    if not project_id.isdecimal():
+        st.warning("Please enter your Project ID correctly.")
+        st.stop()
+    
+    if not api_key.isascii():
+        st.warning("Please enter your API key correctly.")
+        st.stop()
+    
+    st.session_state.api_key = api_key
+    st.session_state.project_id = int(project_id)
+    client = Prem(api_key=api_key)
+    st.rerun()
 else:
     client = Prem(api_key=st.session_state.api_key)
 
@@ -65,9 +70,9 @@ def get_ai_response(messages):
             st.button("Stop generating")
 
         shown_message = ""
-        
+
         for chunk in stream:
-            if chunk.choices[0].delta["content"]:
+            if chunk.choices and chunk.choices[0].delta.get("content"):
                 content = chunk.choices[0].delta["content"]
                 st.session_state.response += content
                 shown_message += content.replace("\n", "  \n")