Update app.py

app.py

@@ -1,8 +1,18 @@
+import os
 import gradio as gr
 from gradio_client import Client
 
-# Initialize the Client for the Hugging Face Space
-client = Client("segestic/token-cost-calculator")
+# Access the Hugging Face token from environment variables
+hf_token = os.getenv("HF_TOKEN")
+
+# Check if the token is set (optional for debugging)
+if hf_token is None:
+    print("Hugging Face token not found. Please set the HF_TOKEN environment variable.")
+else:
+    print("Hugging Face token successfully retrieved.")
+
+# Initialize the Client for the Hugging Face Space with the token
+client = Client("segestic/token-cost-calculator", token=hf_token)
 
 # Function to interact with the Hugging Face Space API
 def interact_with_space(system_prompt, user_prompt, assistant_response, model="gpt-4o"):