Update app.py

app.py

@@ -135,10 +135,13 @@ def upload_parquet(file_path):
 
 def text_to_parquet(text):
     try:
-        # 텍스트를 DataFrame으로 변환 (예시: 각 줄을 새로운 행으로)
+        # 텍스트를 DataFrame으로 변환 (예시: 각 줄을 새로운 행으로, 콤마로 구분)
         data = [line.split(',') for line in text.strip().split('\n')]
         df = pd.DataFrame(data, columns=['id', 'text', 'label', 'metadata'])
 
+        # 데이터 유형 최적화
+        df = df.astype({'id': 'int32', 'text': 'string', 'label': 'category', 'metadata': 'string'})
+
         # Parquet 파일로 변환
         parquet_filename = 'text_to_parquet.parquet'
         df.to_parquet(parquet_filename, engine='pyarrow', compression='snappy')
@@ -150,7 +153,7 @@ def text_to_parquet(text):
         with open(parquet_filename, "rb") as f:
             data = f.read()
 
-        return f"{parquet_filename} 파일이 성공적으로 변환되었습니다.", parquet_content, (parquet_filename, data)
+        return f"{parquet_filename} 파일이 성공적으로 변환되었습니다.", parquet_content, data
     except Exception as e:
         return f"텍스트 변환 중 오류가 발생했습니다: {str(e)}", "", None
 
@@ -158,10 +161,18 @@ css = """
 footer {
     visibility: hidden;
 }
-.chatbot-container {
+#chatbot-container {
     height: 600px;
+    overflow-y: scroll;
 }
-.chatbot-container .message {
+#chatbot-container .message {
+    font-size: 14px;
+}
+#chatbot-data-upload {
+    height: 600px;
+    overflow-y: scroll;
+}
+#chatbot-data-upload .message {
     font-size: 14px;
 }
 """
@@ -173,7 +184,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     with gr.Tab("챗봇"):
         gr.Markdown("### LLM과 대화하기")
         chatbot = gr.Chatbot(label="챗봇", elem_id="chatbot-container")
-        msg = gr.Textbox(label="메시지 입력")
+        msg = gr.Textbox(label="메시지 입력", placeholder="여기에 메시지를 입력하세요...")
         send = gr.Button("전송")
 
         with gr.Accordion("시스템 프롬프트 및 옵션 설정", open=False):
@@ -204,6 +215,25 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             outputs=[chatbot, msg]
         )
 
+        # 예제 버튼 (복원된 examples)
+        with gr.Accordion("예제", open=False):
+            example_buttons = gr.Examples(
+                examples=[
+                    ["패션 코드 실행"],
+                    ["UHD 이미지 코드 실행"],
+                    ["MixGEN 코드 실행"],
+                    ["test.parquet 실행"],
+                    ["상세한 사용 방법을 마치 화면을 보면서 설명하듯이 4000 토큰 이상 자세히 설명하라"],
+                    ["FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
+                    ["사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
+                    ["본 서비스를 SEO 최적화하여 블로그 포스트(배경 및 필요성, 기존 유사 서비스와 비교하여 특장점, 활용처, 가치, 기대효과, 결론을 포함)로 4000 토큰 이상 작성하라"],
+                    ["특허 출원에 활용할 기술 및 비즈니스모델 측면을 포함하여 특허 출원서 구성에 맞게 혁신적인 창의 발명 내용을 중심으로 4000 토큰 이상 작성하라."],
+                    ["계속 이어서 답변하라"],
+                ],
+                inputs=msg,
+                label="예제 선택",
+            )
+
     with gr.Tab("데이터 변환"):
         gr.Markdown("### CSV 파일 업로드 및 Parquet 변환")
         with gr.Row():
@@ -221,7 +251,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                         # 파일을 다운로드할 수 있도록 바이너리 데이터로 읽기
                         with open(parquet_filename, "rb") as f:
                             data = f.read()
-                        return message, load_parquet(parquet_filename), (parquet_filename, data)
+                        return message, load_parquet(parquet_filename), data
                     else:
                         return message, "", None
 
@@ -248,9 +278,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 def handle_parquet_upload(file_path):
                     message, parquet_content, parquet_json = upload_parquet(file_path)
                     if parquet_json:
-                        return message, parquet_preview_chat.update(value=parquet_content), parquet_data_state.update(value=parquet_json)
+                        return message, parquet_content, parquet_json
                     else:
-                        return message, gr.Markdown.update(value=""), parquet_data_state.update(value=None)
+                        return message, "", None
 
                 parquet_upload_button.click(
                     handle_parquet_upload,
@@ -260,12 +290,13 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         
         gr.Markdown("### LLM과 대화하기")
         chatbot_data_upload = gr.Chatbot(label="챗봇 데이터 업로드", elem_id="chatbot-data-upload")
-        msg_data_upload = gr.Textbox(label="메시지 입력")
+        msg_data_upload = gr.Textbox(label="메시지 입력", placeholder="여기에 메시지를 입력하세요...")
         send_data_upload = gr.Button("전송")
 
         # 챗봇 메시지 처리 함수 (데이터 업로드 버전)
         def handle_message_data_upload(message, history, system_message, max_tokens, temperature, top_p, parquet_data):
-            # Update history with user message
+            # Parquet 데이터가 업로드된 경우, 추가적인 로직을 구현할 수 있음
+            # 현재는 기본적으로 메시지를 처리
             history = history or []
             history.append({"role": "user", "content": message})
             # Generate response
@@ -289,12 +320,13 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         gr.Markdown("### 텍스트를 입력하면 CSV로 변환 후 Parquet으로 자동 전환됩니다.")
         with gr.Row():
             with gr.Column():
-                text_input = gr.Textbox(label="텍스트 입력 (각 행은 `id,text,label,metadata` 형식으로 입력)", lines=10)
+                text_input = gr.Textbox(label="텍스트 입력 (각 행은 `id,text,label,metadata` 형식으로 입력)", lines=10, placeholder="예: 1,Sample Text,Label1,Metadata1\n2,Another Text,Label2,Metadata2")
                 convert_button = gr.Button("변환 및 다운로드")
                 convert_status = gr.Textbox(label="변환 상태", interactive=False)
                 parquet_preview_convert = gr.Markdown(label="Parquet 파일 미리보기")
                 download_parquet_convert = gr.File(label="Parquet 파일 다운로드", type="binary", interactive=False)
 
+                # 변환 버튼 클릭 시 실행할 함수
                 def handle_text_to_parquet(text):
                     message, parquet_content, file_data = text_to_parquet(text)
                     if file_data:
@@ -322,3 +354,4 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
 if __name__ == "__main__":
     demo.launch()
+