Update app.py

app.py

@@ -25,13 +25,23 @@ def get_pdf_text(pdf_docs):
 # 과제
 # 아래 텍스트 추출 함수를 작성
 def get_text_file(docs):
-    text_list = []  # 텍스트를 저장할 리스트
+    # 텍스트 생성을 위한 Hugging Face의 텍스트 생성 파이프라인 생성
+    #text_generator = pipeline("text-generation", model="distilgpt2")
 
+    # 텍스트 파일로부터 텍스트를 읽어와 리스트에 저장합니다.
+    text_list = []
     for doc in docs:
         content = doc.getvalue().decode("utf-8")  # 파일 내용을 가져와서 UTF-8로 디코딩하여 문자열로 변환
         text_list.append(content)  # 텍스트를 리스트에 추가
 
-    return text_list  # 텍스트가 담긴 리스트 반환
+    # 각 텍스트에 대해 허깅 페이스 모델을 사용하여 텍스트를 생성하고 결과를 저장합니다.
+    generated_texts = []
+    for text in text_list:
+        generated_text = text_generator(text, max_length=50, num_return_sequences=1)
+        generated_texts.append(generated_text[0]['generated_text'])
+
+    return generated_texts  # 생성된 텍스트 리스트 반환
+
     
 def get_csv_file(docs):
     pass