Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 import pandas as pd
-from typing import List, Tuple
+from typing import List, Dict
 
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
@@ -37,7 +37,7 @@ test_parquet_content = load_parquet('test.parquet')
 
 def respond(
     message,
-    history: List[dict],
+    history: List[Dict[str, str]],
     system_message="",  # 기본값 추가
     max_tokens=4000,    # 기본값 변경
     temperature=0.7,    # 기본값 유지
@@ -61,7 +61,10 @@ def respond(
         parquet_content = ""
         for item in history:
             if item['role'] == 'assistant' and 'test.parquet 파일 내용' in item['content']:
-                parquet_content = item['content'].split("```markdown\n")[1].split("\n```")[0]
+                try:
+                    parquet_content = item['content'].split("```markdown\n")[1].split("\n```")[0]
+                except IndexError:
+                    parquet_content = ""
                 break
         system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{parquet_content}\n```"
         message = "test.parquet 파일에 대한 내용을 학습하였고, 관련 설명 및 Q&A를 진행할 준비가 되어있다. 궁금한 점이 있으면 물어보라."
@@ -153,7 +156,7 @@ def text_to_parquet(text):
         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, data)
     except Exception as e:
         return f"텍스트 변환 중 오류가 발생했습니다: {str(e)}", "", None
 
@@ -161,18 +164,11 @@ css = """
 footer {
     visibility: hidden;
 }
-#chatbot-container {
+#chatbot-container, #chatbot-data-upload {
     height: 600px;
     overflow-y: scroll;
 }
-#chatbot-container .message {
-    font-size: 14px;
-}
-#chatbot-data-upload {
-    height: 600px;
-    overflow-y: scroll;
-}
-#chatbot-data-upload .message {
+#chatbot-container .message, #chatbot-data-upload .message {
     font-size: 14px;
 }
 """
@@ -183,7 +179,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")
+        chatbot = gr.Chatbot(label="챗봇", type="messages", elem_id="chatbot-container")
         msg = gr.Textbox(label="메시지 입력", placeholder="여기에 메시지를 입력하세요...")
         send = gr.Button("전송")
 
@@ -251,7 +247,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), data
+                        return message, load_parquet(parquet_filename), (parquet_filename, data)
                     else:
                         return message, "", None
 
@@ -289,13 +285,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")
+        chatbot_data_upload = gr.Chatbot(label="챗봇 데이터 업로드", type="messages", elem_id="chatbot-data-upload")
         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):
-            # Parquet 데이터가 업로드된 경우, 추가적인 로직을 구현할 수 있음
+            # Parquet 데이터를 활용할 수 있는 로직을 추가 (예: 데이터 분석, 질의응답)
             # 현재는 기본적으로 메시지를 처리
             history = history or []
             history.append({"role": "user", "content": message})
@@ -353,5 +349,4 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     gr.Markdown("### Gradio 인터페이스를 사용하여 LLM 모델과 상호작용하세요!")
 
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch(share=True)