Fix API key inputs

README.md

@@ -51,7 +51,7 @@ STREAMLIT_BROWSER_SERVER_ADDRESS=adamelliotfields-api-inference.hf.space
 ```bash
 FAL_KEY=...
 HF_TOKEN=...
-PPLX_API_KEY=...
+PERPLEXITY_API_KEY=...
 ```
 
 ## Installation

pages/1_💬_Text_Generation.py

@@ -5,9 +5,10 @@ import streamlit as st
 
 from lib import Config, ServicePresets, txt2txt_generate
 
-HF_TOKEN = os.environ.get("HF_TOKEN") or None
-PPLX_API_KEY = os.environ.get("PPLX_API_KEY") or None
-
+HF_TOKEN = None
+PERPLEXITY_API_KEY = None
+# HF_TOKEN = os.environ.get("HF_TOKEN") or None
+# PERPLEXITY_API_KEY = os.environ.get("PERPLEXITY_API_KEY") or None
 
 # config
 st.set_page_config(
@@ -51,9 +52,9 @@ if service == "Huggingface" and HF_TOKEN is None:
         value=st.session_state.api_key_huggingface,
     )
 else:
-    st.session_state.api_key_huggingface = None
+    st.session_state.api_key_huggingface = st.session_state.api_key_huggingface
 
-if service == "Perplexity" and PPLX_API_KEY is None:
+if service == "Perplexity" and PERPLEXITY_API_KEY is None:
     st.session_state.api_key_perplexity = st.sidebar.text_input(
         "API Key",
         type="password",
@@ -62,13 +63,13 @@ if service == "Perplexity" and PPLX_API_KEY is None:
         value=st.session_state.api_key_perplexity,
     )
 else:
-    st.session_state.api_key_perplexity = None
+    st.session_state.api_key_perplexity = st.session_state.api_key_perplexity
 
 if service == "Huggingface" and HF_TOKEN is not None:
     st.session_state.api_key_huggingface = HF_TOKEN
 
-if service == "Perplexity" and PPLX_API_KEY is not None:
-    st.session_state.api_key_perplexity = PPLX_API_KEY
+if service == "Perplexity" and PERPLEXITY_API_KEY is not None:
+    st.session_state.api_key_perplexity = PERPLEXITY_API_KEY
 
 model = st.sidebar.selectbox(
     "Model",

pages/2_🎨_Text_to_Image.py

@@ -5,8 +5,10 @@ import streamlit as st
 
 from lib import Config, ModelPresets, txt2img_generate
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
-FAL_KEY = os.environ.get("FAL_KEY")
+HF_TOKEN = None
+FAL_KEY = None
+# HF_TOKEN = os.environ.get("HF_TOKEN") or None
+# FAL_KEY = os.environ.get("FAL_KEY") or None
 API_URL = "https://api-inference.huggingface.co/models"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}", "X-Wait-For-Model": "true", "X-Use-Cache": "false"}
 PRESET_MODEL = {
@@ -54,10 +56,10 @@ if service == "Huggingface" and HF_TOKEN is None:
         type="password",
         help="Cleared on page refresh",
         value=st.session_state.api_key_huggingface,
-        disabled=st.session_state.txt2txt_running,
+        disabled=st.session_state.txt2img_running,
     )
 else:
-    st.session_state.api_key_huggingface = None
+    st.session_state.api_key_huggingface = st.session_state.api_key_huggingface
 
 if service == "Fal" and FAL_KEY is None:
     st.session_state.api_key_fal = st.sidebar.text_input(
@@ -65,10 +67,10 @@ if service == "Fal" and FAL_KEY is None:
         type="password",
         help="Cleared on page refresh",
         value=st.session_state.api_key_fal,
-        disabled=st.session_state.txt2txt_running,
+        disabled=st.session_state.txt2img_running,
     )
 else:
-    st.session_state.api_key_fal = None
+    st.session_state.api_key_fal = st.session_state.api_key_fal
 
 if service == "Huggingface" and HF_TOKEN is not None:
     st.session_state.api_key_huggingface = HF_TOKEN