Update app.py

app.py

@@ -42,18 +42,18 @@ def respond(
     top_p: float = 0.9,
 ) -> str:
     # 시스템 프롬프트 설정
-    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
+    system_prefix = """반드시 한글로 답변할 것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
 
     # 특정 명령어 처리
     if message.lower() == "패션 코드 실행":
         system_prefix += f"\n\n패션 코드 내용:\n```python\n{fashion_code}\n```"
-        message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://aiqcamp-fash.hf.space)을 통해 테스트 해보라고 출력하라."
+        message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://aiqcamp-fash.hf.space)을 통해 테스트해보라고 출력하라."
     elif message.lower() == "uhd 이미지 코드 실행":
         system_prefix += f"\n\nUHD 이미지 코드 내용:\n```python\n{uhdimage_code}\n```"
-        message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-ultpixgen.hf.space)을 통해 테스트 해보라고 출력하라."
+        message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-ultpixgen.hf.space)을 통해 테스트해보라고 출력하라."
     elif message.lower() == "mixgen 코드 실행":
         system_prefix += f"\n\nMixGEN 코드 내용:\n```python\n{MixGEN_code}\n```"
-        message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트 해보라고 출력하라."
+        message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트해보라고 출력하라."
     elif message.lower() == "test.parquet 실행":
         # history에서 parquet_content 찾기
         parquet_content = ""
@@ -119,7 +119,7 @@ def upload_csv(file_path: str) -> Tuple[str, str]:
         # Parquet 파일 내용 미리보기
         parquet_content = load_parquet(parquet_filename)
 
-        return f"{parquet_filename} 파일이 성공적으로 업로드되고 변환되었습니다.", parquet_content
+        return f"{parquet_filename} 파일이 성공적으로 업로드되고 변환되었습니다.", parquet_filename
     except Exception as e:
         return f"CSV 파일 업로드 및 변환 중 오류가 발생했습니다: {str(e)}", ""
 
@@ -129,13 +129,13 @@ def upload_parquet(file_path: str) -> Tuple[str, str, Dict]:
         df = pd.read_parquet(file_path, engine='pyarrow')
 
         # Markdown으로 변환하여 미리보기
-        parquet_content = df.to_markdown(index=False)
+        parquet_content = df.head(10).to_markdown(index=False)
 
         return "Parquet 파일이 성공적으로 업로드되었습니다.", parquet_content, df.to_json()
     except Exception as e:
         return f"Parquet 파일 업로드 중 오류가 발생했습니다: {str(e)}", "", {}
 
-def text_to_parquet(text: str) -> Tuple[str, str, bytes]:
+def text_to_parquet(text: str) -> Tuple[str, str, str]:
     try:
         # 텍스트를 DataFrame으로 변환 (각 행은 콤마로 구분)
         data = [line.split(',') for line in text.strip().split('\n')]
@@ -151,13 +151,9 @@ def text_to_parquet(text: str) -> Tuple[str, str, bytes]:
         # Parquet 파일 내용 미리보기
         parquet_content = load_parquet(parquet_filename)
 
-        # 파일 데이터를 바이너리로 읽기
-        with open(parquet_filename, "rb") as f:
-            data = f.read()
-
-        return f"{parquet_filename} 파일이 성공적으로 변환되었습니다.", parquet_content, data
+        return f"{parquet_filename} 파일이 성공적으로 변환되었습니다.", parquet_content, parquet_filename
     except Exception as e:
-        return f"텍스트 변환 중 오류가 발생했습니다: {str(e)}", "", b""
+        return f"텍스트 변환 중 오류가 발생했습니다: {str(e)}", "", ""
 
 # CSS 설정
 css = """
@@ -240,16 +236,13 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 upload_button = gr.Button("업로드 및 변환")
                 upload_status = gr.Textbox(label="업로드 상태", interactive=False)
                 parquet_preview = gr.Markdown(label="Parquet 파일 미리보기")
-                download_button = gr.Download(label="Parquet 파일 다운로드")
+                download_button = gr.File(label="Parquet 파일 다운로드", interactive=False)
 
-                def handle_csv_upload(file_path: str) -> Tuple[str, str, Tuple[bytes, str]]:
+                def handle_csv_upload(file_path: str):
                     message, parquet_filename = upload_csv(file_path)
                     if parquet_filename:
-                        # Parquet 파일 읽기
-                        with open(parquet_filename, "rb") as f:
-                            data = f.read()
-                        filename = os.path.basename(parquet_filename)
-                        return message, parquet_filename, (data, filename)
+                        parquet_content = load_parquet(parquet_filename)
+                        return message, parquet_content, parquet_filename
                     else:
                         return message, "", None
 
@@ -273,7 +266,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 parquet_preview_chat = gr.Markdown(label="Parquet 파일 미리보기")
                 parquet_data_state = gr.State()
 
-                def handle_parquet_upload(file_path: str) -> Tuple[str, str, Dict]:
+                def handle_parquet_upload(file_path: str):
                     message, parquet_content, parquet_json = upload_parquet(file_path)
                     if parquet_json:
                         return message, parquet_content, parquet_json
@@ -291,7 +284,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         msg_data_upload = gr.Textbox(label="메시지 입력", placeholder="여기에 메시지를 입력하세요...")
         send_data_upload = gr.Button("전송")
 
-        def handle_message_data_upload(message: str, history: List[Dict[str, str]], system_message: str, max_tokens: int, temperature: float, top_p: float, parquet_data: Dict) -> Tuple[List[Dict[str, str]], str]:
+        def handle_message_data_upload(message: str, history: List[Dict[str, str]], system_message: str, max_tokens: int, temperature: float, top_p: float, parquet_data: Dict):
             history = history or []
             history.append({"role": "user", "content": message})
             try:
@@ -312,8 +305,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             outputs=[chatbot_data_upload, msg_data_upload]
         )
 
-    # 네 번째 탭: 텍스트 to csv to parquet 변화
-    with gr.Tab("텍스트 to csv to parquet 변화"):
+    # 네 번째 탭: 텍스트 to csv to parquet 변환
+    with gr.Tab("텍스트 to csv to parquet 변환"):
         gr.Markdown("### 텍스트를 입력하면 CSV로 변환 후 Parquet으로 자동 전환됩니다.")
         with gr.Row():
             with gr.Column():
@@ -325,13 +318,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 convert_button = gr.Button("변환 및 다운로드")
                 convert_status = gr.Textbox(label="변환 상태", interactive=False)
                 parquet_preview_convert = gr.Markdown(label="Parquet 파일 미리보기")
-                download_parquet_convert = gr.Download(label="Parquet 파일 다운로드")
+                download_parquet_convert = gr.File(label="Parquet 파일 다운로드", interactive=False)
 
-                def handle_text_to_parquet(text: str) -> Tuple[str, str, Tuple[bytes, str]]:
-                    message, parquet_content, file_data = text_to_parquet(text)
-                    if file_data:
-                        filename = 'text_to_parquet.parquet'
-                        return message, parquet_content, (file_data, filename)
+                def handle_text_to_parquet(text: str):
+                    message, parquet_content, parquet_filename = text_to_parquet(text)
+                    if parquet_filename:
+                        return message, parquet_content, parquet_filename
                     else:
                         return message, "", None
 
@@ -355,5 +347,5 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     gr.Markdown("### Gradio 인터페이스를 사용하여 LLM 모델과 상호작용하세요!")
 
 if __name__ == "__main__":
-    # share=True는 Hugging Face Spaces에서 지원되지 않으므로 제거
     demo.launch()
+