Update app.py
Browse files
app.py
CHANGED
@@ -26,24 +26,13 @@ def get_pdf_text(pdf_docs):
|
|
26 |
# μλ ν
μ€νΈ μΆμΆ ν¨μλ₯Ό μμ±
|
27 |
|
28 |
def get_text_file(text_docs):
|
29 |
-
# μμ λλ ν 리λ₯Ό μμ±ν©λλ€.
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
with open(temp_filepath, "w", encoding="ascii", errors="ignore") as f:
|
37 |
-
f.write(text_docs.getvalue())
|
38 |
-
|
39 |
-
# μμ νμΌμμ ν
μ€νΈλ₯Ό μ½μ΄μ΅λλ€.
|
40 |
-
with open(temp_filepath, "r", encoding="ascii", errors="ignore") as f:
|
41 |
-
text_content = f.read()
|
42 |
-
|
43 |
-
# μμ λλ ν 리λ₯Ό μλμΌλ‘ μ 리ν©λλ€.
|
44 |
-
temp_dir.cleanup()
|
45 |
-
|
46 |
-
return [text_content] # μΆμΆν ν
μ€νΈλ₯Ό 리μ€νΈμ λ΄μ λ°νν©λλ€.
|
47 |
|
48 |
|
49 |
|
|
|
26 |
# μλ ν
μ€νΈ μΆμΆ ν¨μλ₯Ό μμ±
|
27 |
|
28 |
def get_text_file(text_docs):
|
29 |
+
temp_dir = tempfile.TemporaryDirectory() # μμ λλ ν 리λ₯Ό μμ±ν©λλ€.
|
30 |
+
temp_filepath = os.path.join(temp_dir.name, text_docs.name) # μμ νμΌ κ²½λ‘λ₯Ό μμ±ν©λλ€.
|
31 |
+
with open(temp_filepath, "wb") as f: # μμ νμΌμ λ°μ΄λ리 μ°κΈ° λͺ¨λλ‘ μ½λλ€.
|
32 |
+
f.write(text_docs.getvalue()) # PDF λ¬Έμμ λ΄μ©μ μμ νμΌμ μλλ€.
|
33 |
+
text_loader = TextLoader(temp_filepath) # PyPDFLoaderλ₯Ό μ¬μ©ν΄ PDFλ₯Ό λ‘λν©λλ€.
|
34 |
+
text_doc = text_loader.load() # ν
μ€νΈλ₯Ό μΆμΆν©λλ€.
|
35 |
+
return text_doc # μΆμΆν ν
μ€νΈλ₯Ό λ°νν©λλ€.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
|
37 |
|
38 |
|