Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
| 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)
|