Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import pandas as pd
 from typing import List, Dict, Tuple
 import json
+import io
 
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient(
@@ -39,17 +40,18 @@ def respond(
 ) -> str:
     # 시스템 프롬프트 설정
     if parquet_data:
-        # Parquet 데이터가 업로드된 경우
         system_prefix = """반드시 한글로 답변할 것. 너는 업로드된 데이터를 기반으로 질문에 답변하는 역할을 한다. 데이터를 분석하여 사용자에게 도움이 되는 정보를 제공하라. 데이터를 활용하여 상세하고 정확한 답변을 제공하되, 민감한 정보나 개인 정보를 노출하지 마라."""
-        # Parquet 데이터를 문자열로 변환하여 시스템 메시지에 포함
-        # 데이터가 너무 클 경우 토큰 제한이 있으므로, 주요 내용을 요약하여 포함하거나 데이터의 구조와 특징을 설명하도록 한다.
-        df = pd.read_json(parquet_data)
+        
+        # 수정된 부분: StringIO로 감싸서 데이터프레임 생성
+        df = pd.read_json(io.StringIO(parquet_data))
+        
         # 데이터의 요약 정보 생성
         data_summary = df.describe(include='all').to_string()
         system_prefix += f"\n\n업로드된 데이터의 요약 정보:\n{data_summary}"
     else:
-        # Parquet 데이터가 업로드되지 않은 경우
         system_prefix = system_message or "너는 AI 조언자 역할이다."
+        
+
 
     # 시스템 메시지와 대화 기록 결합
     messages = [{"role": "system", "content": system_prefix}]