Update app.py

app.py

@@ -26,86 +26,19 @@ 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)
-client = weaviate.Client(url=weaviate_url, auth_client_secret=auth_config, 
-                         additional_headers={"X-Cohere-Api-Key": cohere_api_key})
+# Function to refresh authentication
+def refresh_authentication():
+    global my_credentials, client
+    my_credentials = weaviate.auth.AuthClientPassword(username=weaviate_username, password=weaviate_password)
+    client = weaviate.Client(weaviate_url, auth_client_secret=my_credentials)
+
+# Initialize the scheduler for authentication refresh
+scheduler = BackgroundScheduler()
+scheduler.add_job(refresh_authentication, 'interval', minutes=30)
+scheduler.start()
+
+# Initial authentication
+refresh_authentication()
 
 # Initialize vectorstore
 vectorstore = Weaviate(client, index_name="HereChat", text_key="text")