Update app.py

app.py

@@ -28,6 +28,8 @@ text_models = []
 free_text_models = []
 embedding_models = []
 free_embedding_models = []
+image_models = []
+free_image_models = []
 
 invalid_keys_global = []
 free_keys_global = []
@@ -100,11 +102,14 @@ def refresh_models():
     """
     global text_models, free_text_models
     global embedding_models, free_embedding_models
+    global image_models, free_image_models
 
     text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
     embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
+    image_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
     free_text_models = []
     free_embedding_models = []
+    free_image_models = []
 
     ban_models_str = os.environ.get("BAN_MODELS")
     ban_models = []
@@ -124,6 +129,7 @@ def refresh_models():
     
     text_models = [model for model in text_models if model not in ban_models]
     embedding_models = [model for model in embedding_models if model not in ban_models]
+    image_models = [model for model in embedding_models if model not in ban_models]
 
     with concurrent.futures.ThreadPoolExecutor(
         max_workers=100
@@ -161,11 +167,31 @@ def refresh_models():
                     free_embedding_models.append(model)
             except Exception as exc:
                 logging.error(f"模型 {model} 测试生成异常: {exc}")
+                
+    with concurrent.futures.ThreadPoolExecutor(
+        max_workers=100
+    ) as executor:
+        future_to_model = {
+            executor.submit(
+                test_image_model_availability,
+                FREE_MODEL_TEST_KEY, model
+            ): model for model in embedding_models
+        }
+        for future in concurrent.futures.as_completed(future_to_model):
+            model = future_to_model[future]
+            try:
+                is_free = future.result()
+                if is_free:
+                    free_image_models.append(model)
+            except Exception as exc:
+                logging.error(f"模型 {model} 测试生成异常: {exc}")
 
     logging.info(f"所有文本模型列表：{text_models}")
     logging.info(f"免费文本模型列表：{free_text_models}")
     logging.info(f"所有向量模型列表：{embedding_models}")
     logging.info(f"免费向量模型列表：{free_embedding_models}")
+    logging.info(f"所有生图模型列表：{image_models}")
+    logging.info(f"免费生图模型列表：{free_image_models}")
 
 def test_embedding_model_availability(api_key, model_name):
     """
@@ -195,6 +221,17 @@ def test_embedding_model_availability(api_key, model_name):
             f"API Key：{api_key}，错误信息：{e}"
         )
         return False
+        
+def test_image_model_availability(api_key, model_name):
+    """
+    测试指定的向量模型是否可用。
+    """
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+
+    return True
 
 def load_keys():
     """