Updated API logic

app.py

@@ -20,13 +20,19 @@ if "REPLICATE_API_TOKEN" not in st.session_state:
 with st.sidebar:
     # Get Replicate API Token
     replicate_api = st.secrets.get('REPLICATE_API_TOKEN', '')
-    if not replicate_api or not (replicate_api.startswith('r8_') and len(replicate_api) == 40):
-        replicate_api = st.text_input('Enter your Replicate API token:', type='password')
-        if not replicate_api:
-            st.warning('No API Key yet! You may get one by signing up at Replicate.')
+    input_token = st.text_input('Enter your Replicate API token:', type='password')
+    
+    if input_token:
+        replicate_api = input_token
+
+# Validate the token
+    if not replicate_api:
+        st.warning('No API Key provided! You may get one by signing up at Replicate.')
+    elif not (replicate_api.startswith('r8_') and len(replicate_api) == 40):
+        st.error('Invalid API Key! Please double-check your token.')
     else:
         st.success('Replicate API token is valid!')
-    
+
     # Set environment variable for Replicate
     os.environ['REPLICATE_API_TOKEN'] = replicate_api
     st.session_state['REPLICATE_API_TOKEN'] = replicate_api