Update app.py

app.py

@@ -14,9 +14,42 @@ import numpy as np
 from PIL import Image as PILImage
 import base64
 from io import BytesIO
+import mysql.connector
 
 # Set the base directory
-BASE_DIR = os.getcwd()
+#BASE_DIR = os.getcwd()
+# MySQLに接続
+conn = mysql.connector.connect(
+        host="www.ryhintl.com",
+        user="smairuser",
+        password="smairuser",
+        port=36000,
+        database="smair"
+    )
+
+cursor = conn.cursor(dictionary=True)
+
+def get_api_keys():   
+    select_one_data_query = "SELECT * FROM agentic_apis_copy where api = 'GROQ_KEYS'"
+    cursor.execute(select_one_data_query)
+    result = cursor.fetchall()
+    keys = [item['key'] for item in result]
+    rtnkey = ",".join(map(str, keys))
+    
+    return rtnkey
+
+# List of Groq tokens
+mytokens = get_api_keys()
+
+tokens = eval("["+mytokens+"]")
+
+def get_next_token():
+    token = tokens[random.randint(0, len(tokens) - 1)]
+    return token
+
+token = get_next_token()
+
+os.environ["GROQ_API_KEY"] = token
 
 GROQ_API_KEY = os.environ.get('GROQ_API_KEY')