jeongsoo commited on
Commit
e234e39
Β·
1 Parent(s): ae964c2

Upload fix

Browse files
Files changed (2) hide show
  1. app/app.py +3 -1
  2. app_gradio.py +5 -2
app/app.py CHANGED
@@ -329,6 +329,7 @@ def upload_document():
329
  return jsonify({"error": "λ¬Έμ„œ 파일이 μ œκ³΅λ˜μ§€ μ•Šμ•˜μŠ΅λ‹ˆλ‹€."}), 400
330
 
331
  doc_file = request.files['document']
 
332
 
333
  # 파일λͺ…이 λΉ„μ–΄μžˆλŠ”μ§€ 확인
334
  if doc_file.filename == '':
@@ -336,7 +337,8 @@ def upload_document():
336
 
337
  # 파일 ν˜•μ‹ 확인
338
  if not allowed_doc_file(doc_file.filename):
339
- return jsonify({"error": "ν—ˆμš©λ˜μ§€ μ•ŠλŠ” 파일 ν˜•μ‹μž…λ‹ˆλ‹€."}), 400
 
340
 
341
  # 파일λͺ… λ³΄μ•ˆ 처리
342
  filename = secure_filename(doc_file.filename)
 
329
  return jsonify({"error": "λ¬Έμ„œ 파일이 μ œκ³΅λ˜μ§€ μ•Šμ•˜μŠ΅λ‹ˆλ‹€."}), 400
330
 
331
  doc_file = request.files['document']
332
+ logger.info(f"받은 파일λͺ…: {doc_file.filename}")
333
 
334
  # 파일λͺ…이 λΉ„μ–΄μžˆλŠ”μ§€ 확인
335
  if doc_file.filename == '':
 
337
 
338
  # 파일 ν˜•μ‹ 확인
339
  if not allowed_doc_file(doc_file.filename):
340
+ logger.error(f"ν—ˆμš©λ˜μ§€ μ•ŠλŠ” 파일 ν˜•μ‹: {doc_file.filename}")
341
+ return jsonify({"error": "ν—ˆμš©λ˜μ§€ μ•ŠλŠ” 파일 ν˜•μ‹μž…λ‹ˆλ‹€. ν˜„μž¬ ν—ˆμš©λœ 파일 ν˜•μ‹: {}".format(', '.join(ALLOWED_DOC_EXTENSIONS))}), 400
342
 
343
  # 파일λͺ… λ³΄μ•ˆ 처리
344
  filename = secure_filename(doc_file.filename)
app_gradio.py CHANGED
@@ -87,11 +87,14 @@ with gr.Blocks(title="RAG 검색 챗봇 with μŒμ„±μΈμ‹") as demo:
87
  if not doc_file:
88
  return "λ¬Έμ„œλ₯Ό μ—…λ‘œλ“œν•˜μ„Έμš”."
89
  try:
90
- logger.info("λ¬Έμ„œ μ—…λ‘œλ“œ μš”μ²­: /api/upload")
 
 
 
91
  with open(doc_file, "rb") as f:
92
  response = flask_client.post(
93
  "/api/upload",
94
- data={"document": (f, "document_file")}
95
  )
96
  data = flask_json.loads(response.data)
97
  if "error" in data:
 
87
  if not doc_file:
88
  return "λ¬Έμ„œλ₯Ό μ—…λ‘œλ“œν•˜μ„Έμš”."
89
  try:
90
+ logger.info(f"λ¬Έμ„œ μ—…λ‘œλ“œ μš”μ²­: /api/upload, 파일λͺ…: {doc_file.name}")
91
+ file_extension = os.path.splitext(doc_file.name)[1].lower()
92
+ logger.info(f"파일 ν™•μž₯자: {file_extension}")
93
+
94
  with open(doc_file, "rb") as f:
95
  response = flask_client.post(
96
  "/api/upload",
97
+ data={"document": (f, doc_file.name)}
98
  )
99
  data = flask_json.loads(response.data)
100
  if "error" in data: