Update app.py

app.py

@@ -59,7 +59,6 @@ def respond(
     elif message.lower() == "test.parquet 실행":
         # 현재 Parquet 내용은 상태 변수에서 가져와야 함
         # 여기서는 예시로 'parquet_content'가 history에 있다고 가정
-        # 실제 구현에서는 별도의 상태 관리를 통해 Parquet 데이터를 참조해야 함
         parquet_content = ""
         for item in history:
             if isinstance(item, tuple) and item[0].startswith("Parquet"):
@@ -150,22 +149,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
     with gr.Tab("챗봇"):
         gr.Markdown("### LLM과 대화하기")
-        chat = gr.ChatInterface(
-            fn=respond,
-            examples=[
-                ["패션 코드 실행"],
-                ["UHD 이미지 코드 실행"],
-                ["MixGEN 코드 실행"],
-                ["test.parquet 실행"],
-                ["상세한 사용 방법을 마치 화면을 보면서 설명하듯이 4000 토큰 이상 자세히 설명하라"],
-                ["FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
-                ["사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
-                ["본 서비스를 SEO 최적화하여 블로그 포스트(배경 및 필요성, 기존 유사 서비스와 비교하여 특장점, 활용처, 가치, 기대효과, 결론을 포함)로 4000 토큰 이상 작성하라"],
-                ["특허 출원에 활용할 기술 및 비즈니스모델 측면을 포함하여 특허 출원서 구성에 맞게 혁신적인 창의 발명 내용을 중심으로 4000 토큰 이상 작성하라."],
-                ["계속 이어서 답변하라"],
-            ],
-            theme="default",  # 원하는 테마로 변경 가능
-        )
+        chatbot = gr.Chatbot(label="챗봇")
+        msg = gr.Textbox(label="메시지 입력")
+        send = gr.Button("전송")
 
         with gr.Accordion("시스템 프롬프트 및 옵션 설정", open=False):
             system_message = gr.Textbox(label="System Message", value="너는 AI 조언자 역할이다.")
@@ -173,6 +159,28 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             temperature = gr.Slider(minimum=0, maximum=1, value=0.7, label="Temperature")
             top_p = gr.Slider(minimum=0, maximum=1, value=0.9, label="Top P")
 
+        # 챗봇 메시지 처리 함수
+        def handle_message(message, history, system_message, max_tokens, temperature, top_p):
+            # Update history with user message
+            history = history or []
+            history.append((message, None))
+            # Generate response
+            response = ""
+            try:
+                for token in respond(message, history, system_message, max_tokens, temperature, top_p):
+                    response = token
+                # Update history with assistant response
+                history[-1] = (message, response)
+            except Exception as e:
+                history[-1] = (message, f"추론 중 오류가 발생했습니다: {str(e)}")
+            return history, ""
+
+        send.click(
+            handle_message,
+            inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p],
+            outputs=[chatbot, msg]
+        )
+
     with gr.Tab("데이터 변환"):
         gr.Markdown("### CSV 파일 업로드 및 Parquet 변환")
         with gr.Row():
@@ -181,7 +189,7 @@ 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.File(label="Parquet 파일 다운로드", file=None, interactive=False)
+                download_button = gr.File(label="Parquet 파일 다운로드", type="binary", interactive=False)
 
                 # 업로드 버튼 클릭 시 실행할 함수
                 def handle_csv_upload(file):
@@ -190,7 +198,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
 
@@ -228,21 +236,30 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 )
 
         gr.Markdown("### LLM과 대화하기")
-        chat_interface = gr.ChatInterface(
-            fn=respond,
-            examples=[
-                ["패션 코드 실행"],
-                ["UHD 이미지 코드 실행"],
-                ["MixGEN 코드 실행"],
-                ["test.parquet 실행"],
-                ["상세한 사용 방법을 마치 화면을 보면서 설명하듯이 4000 토큰 이상 자세히 설명하라"],
-                ["FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
-                ["사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
-                ["본 서비스를 SEO 최적화하여 블로그 포스트(배경 및 필요성, 기존 유사 서비스와 비교하여 특장점, 활용처, 가치, 기대효과, 결론을 포함)로 4000 토큰 이상 작성하라"],
-                ["특허 출원에 활용할 기술 및 비즈니스모델 측면을 포함하여 특허 출원서 구성에 맞게 혁신적인 창의 발명 내용을 중심으로 4000 토큰 이상 작성하라."],
-                ["계속 이어서 답변하라"],
-            ],
-            theme="default",  # 원하는 테마로 변경 가능
+        chatbot_data_upload = gr.Chatbot(label="챗봇 데이터 업로드")
+        msg_data_upload = gr.Textbox(label="메시지 입력")
+        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
+            history = history or []
+            history.append((message, None))
+            # Generate response
+            response = ""
+            try:
+                for token in respond(message, history, system_message, max_tokens, temperature, top_p):
+                    response = token
+                # Update history with assistant response
+                history[-1] = (message, response)
+            except Exception as e:
+                history[-1] = (message, f"추론 중 오류가 발생했습니다: {str(e)}")
+            return history, ""
+
+        send_data_upload.click(
+            handle_message_data_upload,
+            inputs=[msg_data_upload, chatbot_data_upload, system_message, max_tokens, temperature, top_p, parquet_data_state],
+            outputs=[chatbot_data_upload, msg_data_upload]
         )
 
     gr.Markdown("## 주의 사항")