Update app.py

app.py

@@ -7,8 +7,12 @@ session = SessionManager()
 ai = AIService()
 
 
-def set_keys(openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider):
-    session.set_keys(openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider)
+def set_keys(
+    openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider, hf_token
+):
+    session.set_keys(
+        openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider, hf_token
+    )
     ok, msg = validate_all_keys(session)
     return msg
 
@@ -48,6 +52,7 @@ with gr.Blocks(theme=gr.themes.Base(default_color_scheme="dark")) as ui:
         kucoin_key = gr.Textbox(label="KuCoin Key", type="password")
         kucoin_secret = gr.Textbox(label="KuCoin Secret", type="password")
         kucoin_pass = gr.Textbox(label="KuCoin Passphrase", type="password")
+        hf_token = gr.Textbox(label="HF Token (if using HuggingFace)", type="password")
         provider = gr.Radio(
             ["OpenAI", "HuggingFace"], value="OpenAI", label="AI Provider"
         )
@@ -55,7 +60,14 @@ with gr.Blocks(theme=gr.themes.Base(default_color_scheme="dark")) as ui:
         status = gr.Textbox(label="Connection Status")
         connect.click(
             set_keys,
-            inputs=[openai_key, kucoin_key, kucoin_secret, kucoin_pass, provider],
+            inputs=[
+                openai_key,
+                kucoin_key,
+                kucoin_secret,
+                kucoin_pass,
+                provider,
+                hf_token,
+            ],
             outputs=status,
         )
         gr.Button("❌ Clear Session").click(clear_keys, outputs=status)