Update app.py

app.py

@@ -144,7 +144,6 @@ def text_to_parquet(text: str) -> Tuple[str, str, str]:
         print(error_message)
         return error_message, "", ""
 
-
 def preprocess_text_with_llm(input_text: str) -> str:
     if not input_text.strip():
         return "입력 텍스트가 비어있습니다."
@@ -162,7 +161,10 @@ def preprocess_text_with_llm(input_text: str) -> str:
 - 텍스트에 쉼표가 있으면 큰따옴표로 감싸기
 - 큰따옴표는 백슬래시로 이스케이프 처리
 - 각 행은 새로운 줄로 구분
-- 모든 필드는 쉼표로 구분"""
+- 모든 필드는 쉼표로 구분
+
+입력 텍스트:
+"""
 
     full_prompt = f"{system_prompt}\n\n{input_text}\n\n출력:"
 
@@ -170,8 +172,8 @@ def preprocess_text_with_llm(input_text: str) -> str:
         response = ""
         stream = hf_client.text_generation(
             prompt=full_prompt,
-            max_new_tokens=4000,
-            temperature=0.3,
+            max_new_tokens=4000,  # 토큰 수 증가
+            temperature=0.3,      # 더 결정적인 출력을 위해 낮춤
             top_p=0.9,
             stream=True,
         )
@@ -185,6 +187,7 @@ def preprocess_text_with_llm(input_text: str) -> str:
         
         # CSV 형식 검증
         try:
+            # StringIO를 사용하여 CSV 형식 검증
             from io import StringIO
             import csv
             csv.reader(StringIO(processed_text))
@@ -193,11 +196,10 @@ def preprocess_text_with_llm(input_text: str) -> str:
             return "LLM이 올바른 CSV 형식을 생성하지 못했습니다. 다시 시도해주세요."
             
     except Exception as e:
-        error_message = f"전처리 중 오류가 발생했습니다: {str(e)}"
+        error_message = f"전처리 중 오류가 발생했습니다: {str(e)}\n{traceback.format_exc()}"
         print(error_message)
         return error_message
 
-
 # CSS 설정
 css = """
 footer {
@@ -495,4 +497,4 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("### [email protected]", elem_id="initial-description")
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)