Update app.py

app.py

@@ -1,5 +1,6 @@
 import weaviate
 import langchain
+import apscheduler
 import gradio as gr
 from langchain.embeddings import CohereEmbeddings
 from langchain.document_loaders import UnstructuredFileLoader
@@ -12,6 +13,8 @@ import ssl
 import mimetypes
 from dotenv import load_dotenv
 import cohere
+from apscheduler.schedulers.background import BackgroundScheduler
+import time
 
 # Load environment variables
 load_dotenv()
@@ -19,6 +22,85 @@ openai_api_key = os.getenv('OPENAI')
 cohere_api_key = os.getenv('COHERE')
 weaviate_api_key = os.getenv('WEAVIATE')
 weaviate_url = os.getenv('WEAVIATE_URL')
+weaviate_username = os.getenv('WEAVIATE_USERNAME')
+weaviate_password = os.getenv('WEAVIATE_PASSWORD')
+
+
+
+def refresh_token():
+    url = weaviate_url
+    # Get Weaviate's OIDC configuration
+    weaviate_open_id_config = requests.get(url + "/v1/.well-known/openid-configuration")
+    if weaviate_open_id_config.status_code == "404":
+        print("Your Weaviate instance is not configured with openid")
+
+    response_json = weaviate_open_id_config.json()
+    client_id = response_json["clientId"]
+    href = response_json["href"]
+
+    # Get the token issuer's OIDC configuration
+    response_auth = requests.get(href)
+
+    if "grant_types_supported" in response_auth.json():
+        # For resource owner password flow
+        assert "password" in response_auth.json()["grant_types_supported"]
+
+        username = "username"  # <-- Replace with the actual username
+        password = "password"  # <-- Replace with the actual password
+
+    # Construct the POST request to send to 'token_endpoint'
+        auth_body = {
+            "grant_type": "password",
+            "client_id": client_id,
+            "username": username,
+            "password": password,
+        }
+        response_post = requests.post(response_auth.json()["token_endpoint"], auth_body)
+        print("Your access_token is:")
+        print(response_post.json()["access_token"])
+    else:
+        # For hybrid flow
+        authorization_url = response_auth.json()["authorization_endpoint"]
+        parameters = {
+            "client_id": client_id,
+            "response_type": "code%20id_token",
+            "response_mode": "fragment",
+            "redirect_url": url,
+            "scope": "openid",
+            "nonce": "abcd",
+        }
+        # Construct 'auth_url'
+        parameter_string = "&".join([key + "=" + item for key, item in parameters.items()])
+        response_auth = requests.get(authorization_url + "?" + parameter_string)
+
+        print("Please visit the following url with your browser to login:")
+        print(authorization_url + "?" + parameter_string)
+        print(
+            "After the login you will be redirected, the token is the 'id_token' parameter of the redirection url."
+        )
+
+        # You could use this regular expression to parse the token
+        resp_txt = "Redirection URL"
+        token = re.search("(?<=id_token=).+(?=&)", resp_txt)[0]
+
+    print("Set as bearer token in the clients to access Weaviate.")
+
+    # Create a scheduler
+    scheduler = BackgroundScheduler()
+
+    # Schedule the token refresh function
+    scheduler.add_job(refresh_token, 'interval', minutes=30)  # Adjust the interval as needed
+
+    # Start the scheduler
+    scheduler.start()
+
+    # Keep the script running
+    try:
+        while True:
+            time.sleep(2)
+    except (KeyboardInterrupt, SystemExit):
+        scheduler.shutdown()
+
 
 # Weaviate connection
 auth_config = weaviate.auth.AuthApiKey(api_key=weaviate_api_key)