ginipick commited on
Commit
6c1d009
ยท
1 Parent(s): ce38851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -26,10 +26,10 @@ if not EMBEDDING_DIR.exists():
26
  EMBEDDING_DIR.mkdir(parents=True)
27
 
28
  # Fireworks AI API ์„ค์ • (VLM ๋ชจ๋ธ)
29
- FIREWORKS_API_KEY = os.getenv("FIREWORKS_API", "")
30
  FIREWORKS_API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
31
  FIREWORKS_VLM_MODEL = "accounts/fireworks/models/qwen3-vl-235b-a22b-instruct"
32
- HAS_VALID_API_KEY = bool(FIREWORKS_API_KEY and FIREWORKS_API_KEY.strip())
33
 
34
  if HAS_VALID_API_KEY:
35
  logger.info("Fireworks AI VLM API ํ‚ค ์„ค์ • ์™„๋ฃŒ")
@@ -266,6 +266,13 @@ async def analyze_pdf_with_vlm_batched(pdf_id: str, force_refresh: bool = False)
266
  # ์ „์ฒด ๋ถ„์„ ๊ฒฐ๊ณผ ํ•ฉ์น˜๊ธฐ
267
  combined_analysis = "\n\n".join(all_analyses)
268
 
 
 
 
 
 
 
 
269
  # ์š”์•ฝ ์ƒ์„ฑ
270
  summary = ""
271
  if combined_analysis:
 
26
  EMBEDDING_DIR.mkdir(parents=True)
27
 
28
  # Fireworks AI API ์„ค์ • (VLM ๋ชจ๋ธ)
29
+ FIREWORKS_API_KEY = os.getenv("FIREWORKS_API", "").strip() # ์ค„๋ฐ”๊ฟˆ/๊ณต๋ฐฑ ์ œ๊ฑฐ
30
  FIREWORKS_API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
31
  FIREWORKS_VLM_MODEL = "accounts/fireworks/models/qwen3-vl-235b-a22b-instruct"
32
+ HAS_VALID_API_KEY = bool(FIREWORKS_API_KEY)
33
 
34
  if HAS_VALID_API_KEY:
35
  logger.info("Fireworks AI VLM API ํ‚ค ์„ค์ • ์™„๋ฃŒ")
 
266
  # ์ „์ฒด ๋ถ„์„ ๊ฒฐ๊ณผ ํ•ฉ์น˜๊ธฐ
267
  combined_analysis = "\n\n".join(all_analyses)
268
 
269
+ # ์„ฑ๊ณตํ•œ ๋ถ„์„์ด ์žˆ๋Š”์ง€ ํ™•์ธ
270
+ successful_analyses = [a for a in all_analyses if "[๋ถ„์„ ์‹คํŒจ:" not in a]
271
+ if not successful_analyses:
272
+ logger.error("๋ชจ๋“  ๋ฐฐ์น˜ ๋ถ„์„ ์‹คํŒจ")
273
+ analysis_status[pdf_id] = {"status": "error", "error": "๋ชจ๋“  ํŽ˜์ด์ง€ ๋ถ„์„ ์‹คํŒจ"}
274
+ return {"error": "PDF ๋ถ„์„์— ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค. API ํ‚ค๋ฅผ ํ™•์ธํ•ด์ฃผ์„ธ์š”."}
275
+
276
  # ์š”์•ฝ ์ƒ์„ฑ
277
  summary = ""
278
  if combined_analysis: