vithacocf commited on
Commit
16dca97
·
verified ·
1 Parent(s): 6f5a769

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -9,9 +9,18 @@ import requests
9
  import fitz # PyMuPDF
10
  import camelot
11
  import pdfplumber
 
12
 
13
  # ================== CONFIG ==================
14
- DEFAULT_API_KEY = "AIzaSyBbK-1P3JD6HPyE3QLhkOps6_-Xo3wUFbs"
 
 
 
 
 
 
 
 
15
 
16
  INTERNAL_MODEL_MAP = {
17
  "Gemini 2.5 Flash": "gemini-2.5-flash",
@@ -252,7 +261,9 @@ def call_gemini_with_prompt(
252
  Ưu tiên: nếu user nhập prompt riêng → dùng prompt đó, ngược lại dùng PROMPT_FREIGHT_JSON.
253
  Header (nếu có) sẽ được chèn thêm vào đầu để giúp model nhận diện hãng bay, thời gian hiệu lực, v.v.
254
  """
255
- api_key = os.environ.get("GOOGLE_API_KEY", DEFAULT_API_KEY)
 
 
256
  genai.configure(api_key=api_key)
257
 
258
  model = genai.GenerativeModel(
@@ -336,7 +347,8 @@ def run_process(file, question, model_choice, temperature, top_p, external_api_u
336
  except Exception as e:
337
  return f"ERROR: {type(e).__name__}: {e}", None
338
  def run_process_internal_base_v2(file_bytes, filename, mime, question, model_choice, temperature, top_p, batch_size=3):
339
- api_key = os.environ.get("GOOGLE_API_KEY", DEFAULT_API_KEY)
 
340
  if not api_key:
341
  return "ERROR: Missing GOOGLE_API_KEY.", None
342
  genai.configure(api_key=api_key)
 
9
  import fitz # PyMuPDF
10
  import camelot
11
  import pdfplumber
12
+ import random
13
 
14
  # ================== CONFIG ==================
15
+ #DEFAULT_API_KEY = "AIzaSyBbK-1P3JD6HPyE3QLhkOps6_-Xo3wUFbs"
16
+ DEFAULT_API_KEY = ["AIzaSyD2FLH3g8cqA1T0CZxETqpkM9O85SW2csA",
17
+ "AIzaSyCRShiCasMPV1FugzPX_3V5LAz-Vjqt8FI",
18
+ "AIzaSyAjnvvAY8if-jGRBu9jpvXKMz8U9V5IRz4",
19
+ "AIzaSyDaWoSpgK8hKiDl6yBpcEow2Tp1bd-V5-I",
20
+ "AIzaSyCsxR162atCCj2ssxiiVa5ejishRbyLDe8",
21
+ "AIzaSyDRWRwwnYJktCULH8d26mzD1Lv4l0CdQws"
22
+ ]
23
+
24
 
25
  INTERNAL_MODEL_MAP = {
26
  "Gemini 2.5 Flash": "gemini-2.5-flash",
 
261
  Ưu tiên: nếu user nhập prompt riêng → dùng prompt đó, ngược lại dùng PROMPT_FREIGHT_JSON.
262
  Header (nếu có) sẽ được chèn thêm vào đầu để giúp model nhận diện hãng bay, thời gian hiệu lực, v.v.
263
  """
264
+
265
+ api_key = random.choice(DEFAULT_API_KEY)
266
+ #os.environ.get("GOOGLE_API_KEY", DEFAULT_API_KEY)
267
  genai.configure(api_key=api_key)
268
 
269
  model = genai.GenerativeModel(
 
347
  except Exception as e:
348
  return f"ERROR: {type(e).__name__}: {e}", None
349
  def run_process_internal_base_v2(file_bytes, filename, mime, question, model_choice, temperature, top_p, batch_size=3):
350
+ #api_key = os.environ.get("GOOGLE_API_KEY", DEFAULT_API_KEY)
351
+ api_key = random.choice(DEFAULT_API_KEY)
352
  if not api_key:
353
  return "ERROR: Missing GOOGLE_API_KEY.", None
354
  genai.configure(api_key=api_key)