Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 import pandas as pd
-from typing import List, Dict
+from typing import List, Dict, Tuple
 
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
-# hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
 
-def load_code(filename):
+def load_code(filename: str) -> str:
     try:
         with open(filename, 'r', encoding='utf-8') as file:
             return file.read()
@@ -17,10 +16,9 @@ def load_code(filename):
     except Exception as e:
         return f"파일을 읽는 중 오류가 발생했습니다: {str(e)}"
 
-def load_parquet(filename):
+def load_parquet(filename: str) -> str:
     try:
         df = pd.read_parquet(filename, engine='pyarrow')
-        # 데이터프레임의 첫 몇 행을 문자열로 변환 (Markdown 표 형식)
         return df.head(10).to_markdown(index=False)
     except FileNotFoundError:
         return f"{filename} 파일을 찾을 수 없습니다."
@@ -36,28 +34,28 @@ MixGEN_code = load_code('mgen.cod')
 test_parquet_content = load_parquet('test.parquet')
 
 def respond(
-    message,
+    message: str,
     history: List[Dict[str, str]],
-    system_message="",  # 기본값 추가
-    max_tokens=4000,    # 기본값 변경
-    temperature=0.7,    # 기본값 유지
-    top_p=0.9,           # 기본값 유지
-    ):
+    system_message: str = "",
+    max_tokens: int = 4000,
+    temperature: float = 0.7,
+    top_p: float = 0.9,
+) -> str:
     # 시스템 프롬프트 설정
     system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
-    
-    # 명령어 처리
+
+    # 특정 명령어 처리
     if message.lower() == "패션 코드 실행":
-        system_message += f"\n\n패션 코드 내용:\n```python\n{fashion_code}\n```"
+        system_prefix += f"\n\n패션 코드 내용:\n```python\n{fashion_code}\n```"
         message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://aiqcamp-fash.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "uhd 이미지 코드 실행":
-        system_message += f"\n\nUHD 이미지 코드 내용:\n```python\n{uhdimage_code}\n```"
+        system_prefix += f"\n\nUHD 이미지 코드 내용:\n```python\n{uhdimage_code}\n```"
         message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-ultpixgen.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "mixgen 코드 실행":
-        system_message += f"\n\nMixGEN 코드 내용:\n```python\n{MixGEN_code}\n```"
+        system_prefix += f"\n\nMixGEN 코드 내용:\n```python\n{MixGEN_code}\n```"
         message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트 해보라고 출력하라."
     elif message.lower() == "test.parquet 실행":
-        # Parquet 내용 참조
+        # history에서 parquet_content 찾기
         parquet_content = ""
         for item in history:
             if item['role'] == 'assistant' and 'test.parquet 파일 내용' in item['content']:
@@ -66,19 +64,20 @@ def respond(
                 except IndexError:
                     parquet_content = ""
                 break
-        system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{parquet_content}\n```"
+        system_prefix += f"\n\ntest.parquet 파일 내용:\n```markdown\n{parquet_content}\n```"
         message = "test.parquet 파일에 대한 내용을 학습하였고, 관련 설명 및 Q&A를 진행할 준비가 되어있다. 궁금한 점이 있으면 물어보라."
     elif message.lower() == "csv 업로드":
         message = "CSV 파일을 업로드하려면 두 번째 탭을 사용하세요."
 
-    # 시스템 메시지와 사용자 메시지 결합
-    messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
+    # 시스템 메시지와 대화 기록 결합
+    messages = [{"role": "system", "content": system_prefix}]
     for chat in history:
         messages.append({"role": chat['role'], "content": chat['content']})
     messages.append({"role": "user", "content": message})
 
     response = ""
     try:
+        # 모델에 메시지 전송 및 응답 받���
         for msg in hf_client.chat_completion(
             messages,
             max_tokens=max_tokens,
@@ -93,18 +92,18 @@ def respond(
     except Exception as e:
         yield f"추론 중 오류가 발생했습니다: {str(e)}"
 
-def upload_csv(file_path):
+def upload_csv(file_path: str) -> Tuple[str, str]:
     try:
-        # CSV 파일 읽기 (구분자 콤마)
+        # CSV 파일 읽기
         df = pd.read_csv(file_path, sep=',')
 
-        # CSV 파일의 컬럼 확인
+        # 필수 컬럼 확인
         required_columns = {'id', 'text', 'label', 'metadata'}
         available_columns = set(df.columns)
         missing_columns = required_columns - available_columns
 
         if missing_columns:
-            return f"CSV 파일에 다음 필수 컬럼이 누락되었습니다: {', '.join(missing_columns)}", None
+            return f"CSV 파일에 다음 필수 컬럼이 누락되었습니다: {', '.join(missing_columns)}", ""
 
         # 데이터 클렌징
         df.drop_duplicates(inplace=True)
@@ -117,28 +116,28 @@ def upload_csv(file_path):
         parquet_filename = os.path.splitext(os.path.basename(file_path))[0] + '.parquet'
         df.to_parquet(parquet_filename, engine='pyarrow', compression='snappy')
 
-        # Parquet 파일 로드
+        # Parquet 파일 내용 미리보기
         parquet_content = load_parquet(parquet_filename)
 
-        return f"{parquet_filename} 파일이 성공적으로 업로드되고 변환되었습니다.", parquet_filename
+        return f"{parquet_filename} 파일이 성공적으로 업로드되고 변환되었습니다.", parquet_content
     except Exception as e:
-        return f"CSV 파일 업로드 및 변환 중 오류가 발생했습니다: {str(e)}", None
+        return f"CSV 파일 업로드 및 변환 중 오류가 발생했습니다: {str(e)}", ""
 
-def upload_parquet(file_path):
+def upload_parquet(file_path: str) -> Tuple[str, str, Dict]:
     try:
         # Parquet 파일 읽기
         df = pd.read_parquet(file_path, engine='pyarrow')
 
-        # 데이터프레임을 Markdown으로 변환
+        # Markdown으로 변환하여 미리보기
         parquet_content = df.to_markdown(index=False)
 
-        return "Parquet 파일이 성공적으로 업로드되었습니다.", parquet_content, df.to_json()  # JSON으로 데이터 저장
+        return "Parquet 파일이 성공적으로 업로드되었습니다.", parquet_content, df.to_json()
     except Exception as e:
-        return f"Parquet 파일 업로드 중 오류가 발생했습니다: {str(e)}", None, None
+        return f"Parquet 파일 업로드 중 오류가 발생했습니다: {str(e)}", "", {}
 
-def text_to_parquet(text):
+def text_to_parquet(text: str) -> Tuple[str, str, bytes]:
     try:
-        # 텍스트를 DataFrame으로 변환 (예시: 각 줄을 새로운 행으로, 콤마로 구분)
+        # 텍스트를 DataFrame으로 변환 (각 행은 콤마로 구분)
         data = [line.split(',') for line in text.strip().split('\n')]
         df = pd.DataFrame(data, columns=['id', 'text', 'label', 'metadata'])
 
@@ -149,17 +148,18 @@ def text_to_parquet(text):
         parquet_filename = 'text_to_parquet.parquet'
         df.to_parquet(parquet_filename, engine='pyarrow', compression='snappy')
 
-        # Parquet 파일 로드
+        # Parquet 파일 내용 미리보기
         parquet_content = load_parquet(parquet_filename)
 
-        # 파일 데이터를 읽어 바이너리로 반환
+        # 파일 데이터를 바이너리로 읽기
         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
+        return f"텍스트 변환 중 오류가 발생했습니다: {str(e)}", "", b""
 
+# CSS 설정
 css = """
 footer {
     visibility: hidden;
@@ -177,6 +177,7 @@ footer {
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     gr.Markdown("# LLM 서비스 인터페이스")
 
+    # 첫 번째 탭: 챗봇
     with gr.Tab("챗봇"):
         gr.Markdown("### LLM과 대화하기")
         chatbot = gr.Chatbot(label="챗봇", type="messages", elem_id="chatbot-container")
@@ -191,18 +192,18 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
         # 챗봇 메시지 처리 함수
         def handle_message(message, history, system_message, max_tokens, temperature, top_p):
-            # Update history with user message
             history = history or []
             history.append({"role": "user", "content": message})
-            # Generate response
-            response = ""
             try:
-                for token in respond(message, history, system_message, max_tokens, temperature, top_p):
-                    response = token
-                # Update history with assistant response
+                # 응답 생성
+                response_gen = respond(message, history, system_message, max_tokens, temperature, top_p)
+                response = ""
+                for partial in response_gen:
+                    response = partial  # 최종 응답
                 history.append({"role": "assistant", "content": response})
             except Exception as e:
-                history.append({"role": "assistant", "content": f"추론 중 오류가 발생했습니다: {str(e)}"})
+                response = f"추론 중 오류가 발생했습니다: {str(e)}"
+                history.append({"role": "assistant", "content": response})
             return history, ""
 
         send.click(
@@ -211,9 +212,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             outputs=[chatbot, msg]
         )
 
-        # 예제 버튼 (복원된 examples)
+        # 예제 복원
         with gr.Accordion("예제", open=False):
-            example_buttons = gr.Examples(
+            gr.Examples(
                 examples=[
                     ["패션 코드 실행"],
                     ["UHD 이미지 코드 실행"],
@@ -230,6 +231,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 label="예제 선택",
             )
 
+    # 두 번째 탭: 데이터 변환
     with gr.Tab("데이터 변환"):
         gr.Markdown("### CSV 파일 업로드 및 Parquet 변환")
         with gr.Row():
@@ -238,16 +240,16 @@ 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 파일 다운로드", type="binary", interactive=False)
+                download_button = gr.Download(label="Parquet 파일 다운로드")
 
-                # 업로드 버튼 클릭 시 실행할 함수
-                def handle_csv_upload(file_path):
+                def handle_csv_upload(file_path: str) -> Tuple[str, str, Tuple[bytes, str]]:
                     message, parquet_filename = upload_csv(file_path)
                     if parquet_filename:
-                        # 파일을 다운로드할 수 있도록 바이너리 데이터로 읽기
+                        # Parquet 파일 읽기
                         with open(parquet_filename, "rb") as f:
                             data = f.read()
-                        return message, load_parquet(parquet_filename), (parquet_filename, data)
+                        filename = os.path.basename(parquet_filename)
+                        return message, parquet_filename, (data, filename)
                     else:
                         return message, "", None
 
@@ -260,6 +262,7 @@ 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("챗봇 데이터 업로드"):
         gr.Markdown("### Parquet 파일 업로드 및 질문하기")
         with gr.Row():
@@ -268,15 +271,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 parquet_upload_button = gr.Button("업로드")
                 parquet_upload_status = gr.Textbox(label="업로드 상태", interactive=False)
                 parquet_preview_chat = gr.Markdown(label="Parquet 파일 미리보기")
-                # 상태를 저장할 Hidden State
                 parquet_data_state = gr.State()
 
-                def handle_parquet_upload(file_path):
+                def handle_parquet_upload(file_path: str) -> Tuple[str, str, Dict]:
                     message, parquet_content, parquet_json = upload_parquet(file_path)
                     if parquet_json:
                         return message, parquet_content, parquet_json
                     else:
-                        return message, "", None
+                        return message, "", {}
 
                 parquet_upload_button.click(
                     handle_parquet_upload,
@@ -289,21 +291,19 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         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 데이터를 활용할 수 있는 로직을 추가 (예: 데이터 분석, 질의응답)
-            # 현재는 기본적으로 메시지를 처리
+        def handle_message_data_upload(message: str, history: List[Dict[str, str]], system_message: str, max_tokens: int, temperature: float, top_p: float, parquet_data: Dict) -> Tuple[List[Dict[str, str]], str]:
             history = history or []
             history.append({"role": "user", "content": message})
-            # Generate response
-            response = ""
             try:
-                for token in respond(message, history, system_message, max_tokens, temperature, top_p):
-                    response = token
-                # Update history with assistant response
+                # Parquet 데이터를 활용할 수 있는 로직 추가 가능
+                response_gen = respond(message, history, system_message, max_tokens, temperature, top_p)
+                response = ""
+                for partial in response_gen:
+                    response = partial
                 history.append({"role": "assistant", "content": response})
             except Exception as e:
-                history.append({"role": "assistant", "content": f"추론 중 오류가 발생했습니다: {str(e)}"})
+                response = f"추론 중 오류가 발생했습니다: {str(e)}"
+                history.append({"role": "assistant", "content": response})
             return history, ""
 
         send_data_upload.click(
@@ -312,21 +312,26 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             outputs=[chatbot_data_upload, msg_data_upload]
         )
 
+    # 네 번째 탭: 텍스트 to csv to parquet 변화
     with gr.Tab("텍스트 to csv to parquet 변화"):
         gr.Markdown("### 텍스트를 입력하면 CSV로 변환 후 Parquet으로 자동 전환됩니다.")
         with gr.Row():
             with gr.Column():
-                text_input = gr.Textbox(label="텍스트 입력 (각 행은 `id,text,label,metadata` 형식으로 입력)", lines=10, placeholder="예: 1,Sample Text,Label1,Metadata1\n2,Another Text,Label2,Metadata2")
+                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)
+                download_parquet_convert = gr.Download(label="Parquet 파일 다운로드")
 
-                # 변환 버튼 클릭 시 실행할 함수
-                def handle_text_to_parquet(text):
+                def handle_text_to_parquet(text: str) -> Tuple[str, str, Tuple[bytes, str]]:
                     message, parquet_content, file_data = text_to_parquet(text)
                     if file_data:
-                        return message, parquet_content, file_data
+                        filename = 'text_to_parquet.parquet'
+                        return message, parquet_content, (file_data, filename)
                     else:
                         return message, "", None
 
@@ -336,6 +341,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                     outputs=[convert_status, parquet_preview_convert, download_parquet_convert]
                 )
 
+    # 주의 사항
     gr.Markdown("## 주의 사항")
     gr.Markdown("""
     - **CSV 업로드**: CSV 파일을 업로드하면 자동으로 Parquet 파일로 변환됩니다. CSV 파일은 반드시 **콤마(`,`)**로 구분되어야 합니다.
@@ -349,4 +355,5 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     gr.Markdown("### Gradio 인터페이스를 사용하여 LLM 모델과 상호작용하세요!")
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    # share=True는 Hugging Face Spaces에서 지원되지 않으므로 제거
+    demo.launch()