Update app.py

app.py

@@ -42,7 +42,7 @@ def respond(
     parquet_data: Dict = None
 ) -> str:
     # 시스템 프롬프트 설정
-    system_prefix = """반드시 한글로 답변할 것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
+    system_prefix = """반드시 한글로 답변할 것. 너는 주어진 데이터를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 Markdown 형식으로 작성하라. 너는 데이터를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해할 만한 내용에 친절하게 알려주도록 하라. 데이터 전체 내용에 대해서는 보안을 유지하고, 민감한 정보는 공개하지 마라."""
 
     # Parquet 데이터 포함
     if parquet_data:
@@ -54,7 +54,7 @@ def respond(
     messages = [{"role": "system", "content": system_prefix}]
     for chat in history:
         messages.append({"role": chat['role'], "content": chat['content']})
-    messages.append({"role": "user", "content": message})
+    # 마지막 메시지는 이미 history에 있으므로 추가하지 않습니다.
 
     try:
         # 모델에 메시지 전송 및 응답 받기
@@ -155,38 +155,13 @@ textarea, input[type="text"] {
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# My RAG: LLM이 나만의 데이터로 학습한 콘텐츠 생성/답변", elem_id="initial-description")
     gr.Markdown(
-        "### 1) 나만의 데이터를 텍스트로 입력하거나 CSV를 업로드하여 Parquet 포맷 데이터셋 자동 변환합니다.\n"
-        "### 2) Parquet 포맷 데이터셋을 업로드하면, LLM이 맞춤 학습 데이터로 활용하여 응답을 시작합니다.\n"
-        "### Tip) '예제'를 통해 다양한 활용 방법을 체험하고 응용해 보세요.",
+        "### 1) 나만의 데이터를 텍스트로 입력 또는 CSV 업로드하여 Parquet 포맷 데이터셋 자동 변환 2) Parquet 포맷 데이터셋을 업로드하면, LLM이 맞춤 학습 데이터로 활용하여 응답\n"
+        "### Tip) '예제'를 통해 다양한 활용 방법을 체험하고 응용해 보세요, 데이터셋 업로드시 미리보기는 10건만 출력",
         elem_id="initial-description"
     )
 
     # 첫 번째 탭: 챗봇 데이터 업로드 (탭 이름 변경: "My 데이터셋+LLM")
     with gr.Tab("My 데이터셋+LLM"):
-        gr.Markdown("### Parquet 파일 업로드 및 질문하기")
-        with gr.Row():
-            with gr.Column():
-                parquet_upload = gr.File(
-                    label="Parquet 파일 업로드", type="filepath", elem_id="parquet-upload-area"
-                )
-                parquet_upload_button = gr.Button("업로드")
-                parquet_upload_status = gr.Textbox(label="업로드 상태", interactive=False)
-                parquet_preview_chat = gr.Markdown(label="Parquet 파일 미리보기")
-                parquet_data_state = gr.State()
-
-                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
-                    else:
-                        return message, "", ""
-
-                parquet_upload_button.click(
-                    handle_parquet_upload,
-                    inputs=parquet_upload,
-                    outputs=[parquet_upload_status, parquet_preview_chat, parquet_data_state]
-                )
-        
         gr.Markdown("### LLM과 대화하기")
         chatbot_data_upload = gr.Chatbot(label="챗봇", type="messages", elem_id="chatbot-data-upload")
         msg_data_upload = gr.Textbox(label="메시지 입력", placeholder="여기에 메시지를 입력하세요...")
@@ -209,6 +184,8 @@ with gr.Blocks(css=css) as demo:
         ):
             history = history or []
             try:
+                # 사용자의 메시지를 히스토리에 추가
+                history.append({"role": "user", "content": message})
                 # 응답 생성
                 response_gen = respond(
                     message, history, system_message, max_tokens, temperature, top_p, parquet_data
@@ -218,16 +195,13 @@ with gr.Blocks(css=css) as demo:
                     partial_response = partial
                     # 대화 내역 업데이트
                     display_history = history + [
-                        {"role": "user", "content": message},
                         {"role": "assistant", "content": partial_response}
                     ]
                     yield display_history, ""
-                # 대화 내역에 추가
-                history.append({"role": "user", "content": message})
+                # 어시스턴트의 응답을 히스토리에 추가
                 history.append({"role": "assistant", "content": partial_response})
             except Exception as e:
                 response = f"추론 중 오류가 발생했습니다: {str(e)}"
-                history.append({"role": "user", "content": message})
                 history.append({"role": "assistant", "content": response})
                 yield history, ""
 
@@ -240,7 +214,7 @@ with gr.Blocks(css=css) as demo:
                 max_tokens,
                 temperature,
                 top_p,
-                parquet_data_state
+                gr.State(),  # parquet_data_state는 아래에서 업데이트됩니다.
             ],
             outputs=[chatbot_data_upload, msg_data_upload],
             queue=True
@@ -251,16 +225,41 @@ with gr.Blocks(css=css) as demo:
             gr.Examples(
                 examples=[
                     ["업로드된 데이터셋에 대해 요약 설명하라."],
-                    ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 본 서비스를 SEO 최적화하여 블로그 포스트(개요, 배경 및 필요성, 기존 유사 제픔/서비스와 비교하여 특장점, 활용처, 가치, 기대효과, 결론을 포함)로 4000 토큰 이상 작성하라"],                        
+                    ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 본 서비스를 SEO 최적화하여 블로그 포스트(개요, 배경 및 필요성, 기존 유사 제품/서비스와 비교하여 특장점, 활용처, 가치, 기대효과, 결론을 포함)로 4000 토큰 이상 작성하라"],
                     ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
                     ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 제품 상세 페이지 형식의 내용을 4000 토큰 이상 자세히 설명하라"],
                     ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
-                    ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 특허 출원에 활용할 기술 및 비즈니스모델 측면을 포함하여 특허 출원서 구성에 맞게 혁신적인 창의 발명 내용을 중심으로 4000 토큰 이상 작성하라."],                   
+                    ["업로드된 데이터셋 파일을 학습 데이터로 활용하여, 특허 출원에 활용할 기술 및 비즈니스 모델 측면을 포함하여 특허 출원서 구성에 맞게 혁신적인 창의 발명 내용을 중심으로 4000 토큰 이상 작성하라."],
                 ],
                 inputs=msg_data_upload,
                 label="예제 선택",
             )
 
+        # Parquet 파일 업로드를 화면 하단으로 이동
+        gr.Markdown("### Parquet 파일 업로드")
+        with gr.Row():
+            with gr.Column():
+                parquet_upload = gr.File(
+                    label="Parquet 파일 업로드", type="filepath", elem_id="parquet-upload-area"
+                )
+                parquet_upload_button = gr.Button("업로드")
+                parquet_upload_status = gr.Textbox(label="업로드 상태", interactive=False)
+                parquet_preview_chat = gr.Markdown(label="Parquet 파일 미리보기")
+                parquet_data_state = gr.State()
+
+                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
+                    else:
+                        return message, "", ""
+
+                parquet_upload_button.click(
+                    handle_parquet_upload,
+                    inputs=parquet_upload,
+                    outputs=[parquet_upload_status, parquet_preview_chat, parquet_data_state]
+                )
+
     # 두 번째 탭: 데이터 변환 (탭 이름 변경: "CSV to My 데이터셋")
     with gr.Tab("CSV to My 데이터셋"):
         gr.Markdown("### CSV 파일 업로드 및 Parquet 변환")
@@ -318,4 +317,3 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-