Update app.py

app.py

@@ -20,8 +20,8 @@ def load_code(filename):
 def load_parquet(filename):
     try:
         df = pd.read_parquet(filename, engine='pyarrow')
-        # 데이터프레임의 첫 몇 행을 문자열로 변환
-        return df.head(10).to_string(index=False)
+        # 데이터프레임의 첫 몇 행을 문자열로 변환 (Markdown 표 형식)
+        return df.head(10).to_markdown(index=False)
     except FileNotFoundError:
         return f"{filename} 파일을 찾을 수 없습니다."
     except Exception as e:
@@ -45,19 +45,19 @@ def respond(
 ):
     global fashion_code, uhdimage_code, MixGEN_code, test_parquet_content
     system_message = system_message or ""
-    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 \"서비스 사용 설명 및 안내, Q&A를 하는 역할이다\". 아주 친절하고 자세하게 4000토큰 이상 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만 한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
+    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
 
     if message.lower() == "패션 코드 실행":
-        system_message += f"\n\n패션 코드 내용:\n{fashion_code}"
+        system_message += f"\n\n패션 코드 내용:\n```python\n{fashion_code}\n```"
         message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://aiqcamp-fash.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "uhd 이미지 코드 실행":
-        system_message += f"\n\nUHD 이미지 코드 내용:\n{uhdimage_code}"
+        system_message += f"\n\nUHD 이미지 코드 내용:\n```python\n{uhdimage_code}\n```"
         message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-ultpixgen.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "mixgen 코드 실행":
-        system_message += f"\n\nMixGEN 코드 내용:\n{MixGEN_code}"
+        system_message += f"\n\nMixGEN 코드 내용:\n```python\n{MixGEN_code}\n```"
         message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "test.parquet 실행":
-        system_message += f"\n\ntest.parquet 파일 내용:\n{test_parquet_content}"
+        system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{test_parquet_content}\n```"
         message = "test.parquet 파일에 대한 내용을 학습하였고, 관련 설명 및 Q&A를 진행할 준비가 되어있다. 궁금한 점이 있으면 물어보라."
 
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
@@ -79,7 +79,7 @@ def respond(
         ):
             token = message.choices[0].delta.get('content', None)
             if token:
-                response += token.strip()
+                response += token
                 yield response
     except Exception as e:
         yield f"추론 중 오류가 발생했습니다: {str(e)}"