Update app.py

app.py

@@ -58,8 +58,14 @@ def respond(
         message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "test.parquet 실행":
         # 현재 Parquet 내용은 상태 변수에서 가져와야 함
-        current_parquet_content = history.get('parquet_content', "")
-        system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{current_parquet_content}\n```"
+        # 여기서는 예시로 'parquet_content'가 history에 있다고 가정
+        # 실제 구현에서는 별도의 상태 관리를 통해 Parquet 데이터를 참조해야 함
+        parquet_content = ""
+        for item in history:
+            if isinstance(item, tuple) and item[0].startswith("Parquet"):
+                parquet_content = item[1]
+                break
+        system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{parquet_content}\n```"
         message = "test.parquet 파일에 대한 내용을 학습하였고, 관련 설명 및 Q&A를 진행할 준비가 되어있다. 궁금한 점이 있으면 물어보라."
     elif message.lower() == "csv 업로드":
         message = "CSV 파일을 업로드하려면 두 번째 탭을 사용하세요."
@@ -67,9 +73,9 @@ def respond(
     # 시스템 메시지와 사용자 메시지 결합
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
     for val in history:
-        if val[0]:
+        if isinstance(val, tuple) and val[0]:
             messages.append({"role": "user", "content": val[0]})
-        if val[1]:
+        if isinstance(val, tuple) and val[1]:
             messages.append({"role": "assistant", "content": val[1]})
     messages.append({"role": "user", "content": message})
 
@@ -162,7 +168,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         )
 
         with gr.Accordion("시스템 프롬프트 및 옵션 설정", open=False):
-            system_message = gr.Textbox(label="System Message", value="너는 AI 조언자 역할이다다")
+            system_message = gr.Textbox(label="System Message", value="너는 AI 조언자 역할이다.")
             max_tokens = gr.Slider(minimum=1, maximum=8000, value=4000, label="Max Tokens")
             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")
@@ -171,7 +177,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         gr.Markdown("### CSV 파일 업로드 및 Parquet 변환")
         with gr.Row():
             with gr.Column():
-                csv_file = gr.File(label="CSV 파일 업로드", type="file")
+                csv_file = gr.File(label="CSV 파일 업로드", type="filepath")
                 upload_button = gr.Button("업로드 및 변환")
                 upload_status = gr.Textbox(label="업로드 상태", interactive=False)
                 parquet_preview = gr.Markdown(label="Parquet 파일 미리보기")
@@ -179,12 +185,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
                 # 업로드 버튼 클릭 시 실행할 함수
                 def handle_csv_upload(file):
-                    message, parquet_filename = upload_csv(file.name)
+                    message, parquet_filename = upload_csv(file)
                     if parquet_filename:
                         # 파일을 다운로드할 수 있도록 경로 설정
                         with open(parquet_filename, "rb") as f:
                             data = f.read()
-                        return message, parquet_preview.update(value=load_parquet(parquet_filename)), gr.File.update(value=(parquet_filename, data))
+                        return message, load_parquet(parquet_filename), (parquet_filename, data)
                     else:
                         return message, "", None
 
@@ -197,11 +203,11 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         gr.Markdown("### 기존 Parquet 파일")
         gr.Markdown(f"**test.parquet 파일 내용:**\n```markdown\n{test_parquet_content}\n```")
 
-    with gr.Tab("챗봇"):
+    with gr.Tab("챗봇 데이터 업로드"):
         gr.Markdown("### Parquet 파일 업로드 및 질문하기")
         with gr.Row():
             with gr.Column():
-                parquet_upload = gr.File(label="Parquet 파일 업로드", type="file")
+                parquet_upload = gr.File(label="Parquet 파일 업로드", type="filepath")
                 parquet_upload_button = gr.Button("업로드")
                 parquet_upload_status = gr.Textbox(label="업로드 상태", interactive=False)
                 parquet_preview_chat = gr.Markdown(label="Parquet 파일 미리보기")
@@ -209,7 +215,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 parquet_data_state = gr.State()
 
                 def handle_parquet_upload(file):
-                    message, parquet_content, parquet_json = upload_parquet(file.name)
+                    message, parquet_content, parquet_json = upload_parquet(file)
                     if parquet_json:
                         return message, parquet_preview_chat.update(value=parquet_content), parquet_data_state.update(value=parquet_json)
                     else:
@@ -245,7 +251,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     - **Parquet 미리보기**: 업로드된 Parquet 파일의 첫 10개 행이 미리보기로 표시됩니다.
     - **LLM과의 대화**: 변환된 Parquet 파일 내용을 기반으로 LLM이 응답을 생성합니다.
     - **Parquet 다운로드**: 변환된 Parquet 파일을 다운로드하려면 변환된 파일 옆의 다운로드 링크를 클릭하세요.
-    - **챗봇 Parquet 업로드**: 챗봇 탭에서 Parquet 파일을 업로드하면 해당 데이터를 기반으로 질문과 답변을 진행할 수 있습니다.
+    - **챗봇 데이터 업로드**: 챗봇 데이터 업로드 탭에서 Parquet 파일을 업로드하면 해당 데이터를 기반으로 질문과 답변을 진행할 수 있습니다.
     """)
 
     gr.Markdown("### Gradio 인터페이스를 사용하여 LLM 모델과 상호작용하세요!")