Update app.py

app.py

@@ -9,15 +9,23 @@ import uuid
 from datetime import datetime, timezone, timedelta
 from dotenv import load_dotenv
 import json
+from huggingface_hub import HfApi
 
-# Load environment variables
-load_dotenv()
+# Load Hugging Face token from the environment variable
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+if GOOGLE_API_KEY is None:
+    raise ValueError("GOOGLE_API_KEY environment variable is not set. Please set it before running the script.")
+
+
+hf_api = HfApi(
+    token= GOOGLE_API_KEY, # Token is not persisted on the machine.
+)
 
 app = Flask(__name__)
 CORS(app)
 
 # Initialize Gemini client
-client = genai.Client(api_key=os.getenv('GOOGLE_API_KEY'))
+client = genai.Client(api_key=hf_api)
 
 # In-memory storage for demo (in production, use a database)
 document_caches = {}
@@ -384,6 +392,7 @@ HTML_TEMPLATE = """
             </div>
         </div>
     </div>
+
     <script>
         let currentCacheId = null;
         
@@ -791,4 +800,4 @@ def delete_cache(cache_id):
 if __name__ == '__main__':
     import os
     port = int(os.environ.get("PORT", 7860))
-    app.run(debug=True, host='0.0.0.0', port= port)
+    app.run(debug=True, host='0.0.0.0', port=port)