Update app.py

app.py

@@ -22,39 +22,25 @@ def read_uploaded_file(file):
     except Exception as e:
         return f"파일을 읽는 중 오류가 발생했습니다: {str(e)}"
 
-def respond(
-    message,
-    history: List[Tuple[str, str]],
-    fashion_file,      # 파일 업로드 입력
-    uhd_file,         # 파일 업로드 입력
-    mixgen_file,      # 파일 업로드 입력
-    parquet_file,     # 파일 업로드 입력
-    system_message="",
-    max_tokens=1024,
-    temperature=0.7,
-    top_p=0.9,
-):
-    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
+def chat(message, history, uploaded_file, system_message="", max_tokens=4000, temperature=0.7, top_p=0.9):
+    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드나 데이터를 기반으로 "서비스 사용 설명 및 안내, Q&A를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 Markdown 형식으로 작성하라. 너는 입력된 내용을 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만한 내용에 친절하게 알려주도록 하라. 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
 
-    if message.lower() == "패션 코드 실행" and fashion_file is not None:
-        fashion_content = read_uploaded_file(fashion_file)
-        system_message += f"\n\n패션 코드 내용:\n```python\n{fashion_content}\n```"
-        message = "패션 가상피팅에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://aiqcamp-fash.hf.space)을 통해 테스트 해보라고 출력하라."
-    
-    elif message.lower() == "uhd 이미지 코드 실행" and uhd_file is not None:
-        uhd_content = read_uploaded_file(uhd_file)
-        system_message += f"\n\nUHD 이미지 코드 내용:\n```python\n{uhd_content}\n```"
-        message = "UHD 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-ultpixgen.hf.space)을 통해 테스트 해보라고 출력하라."
-    
-    elif message.lower() == "mixgen 코드 실행" and mixgen_file is not None:
-        mixgen_content = read_uploaded_file(mixgen_file)
-        system_message += f"\n\nMixGEN 코드 내용:\n```python\n{mixgen_content}\n```"
-        message = "MixGEN3 이미지 생성에 대한 내용을 학습하였고, 설명할 준비가 되어있다고 알리고 서비스 URL(https://openfree-mixgen3.hf.space)을 통해 테스트 해보라고 출력하라."
-    
-    elif message.lower() == "test.parquet 실행" and parquet_file is not None:
-        parquet_content = read_uploaded_file(parquet_file)
-        system_message += f"\n\ntest.parquet 파일 내용:\n```markdown\n{parquet_content}\n```"
-        message = "test.parquet 파일에 대한 내용을 학습하였고, 관련 설명 및 Q&A를 진행할 준비가 되어있다. 궁금한 점이 있으면 물어보라."
+    if uploaded_file:
+        content = read_uploaded_file(uploaded_file)
+        file_extension = os.path.splitext(uploaded_file.name)[1].lower()
+        
+        if file_extension == '.parquet':
+            system_message += f"\n\n파일 내용:\n```markdown\n{content}\n```"
+        else:
+            system_message += f"\n\n파일 내용:\n```python\n{content}\n```"
+            
+        if message == "파일 분석을 시작합니다.":
+            message = """업로드된 파일을 분석하여 다음 내용을 포함하여 상세히 설명하라:
+1. 파일의 주요 목적과 기능
+2. 주요 특징과 구성요소
+3. 활용 방법 및 사용 시나리오
+4. 주의사항 및 제한사항
+5. 기대효과 및 장점"""
 
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
     for val in history:
@@ -66,49 +52,43 @@ def respond(
 
     response = ""
     try:
-        for message in hf_client.chat_completion(
+        for msg in hf_client.chat_completion(
             messages,
             max_tokens=max_tokens,
             stream=True,
             temperature=temperature,
             top_p=top_p,
         ):
-            token = message.choices[0].delta.get('content', None)
+            token = msg.choices[0].delta.get('content', None)
             if token:
                 response += token
-                yield response
+        return "", history + [(message, response)]
     except Exception as e:
-        yield f"추론 중 오류가 발생했습니다: {str(e)}"
+        error_msg = f"추론 중 오류가 발생했습니다: {str(e)}"
+        return "", history + [(message, error_msg)]
 
 css = """
-footer {
-    visibility: hidden;
-}
+footer {visibility: hidden}
 """
 
-
-# ... (이전 import 문과 함수들은 동일)
-
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     with gr.Row():
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot()
-            msg = gr.Textbox(label="메시지를 입력하세요")
+            chatbot = gr.Chatbot(height=600, type="messages")
+            msg = gr.Textbox(
+                label="메시지를 입력하세요",
+                show_label=False,
+                placeholder="메시지를 입력하세요...",
+                container=False
+            )
             clear = gr.ClearButton([msg, chatbot])
         
         with gr.Column(scale=1):
-            with gr.Group():
-                fashion_file = gr.File(label="Fashion Code File", file_types=[".cod", ".txt", ".py"])
-                fashion_analyze = gr.Button("패션 코드 분석")
-                
-                uhd_file = gr.File(label="UHD Image Code File", file_types=[".cod", ".txt", ".py"])
-                uhd_analyze = gr.Button("UHD 이미지 코드 분석")
-                
-                mixgen_file = gr.File(label="MixGEN Code File", file_types=[".cod", ".txt", ".py"])
-                mixgen_analyze = gr.Button("MixGEN 코드 분석")
-                
-                parquet_file = gr.File(label="Parquet File", file_types=[".parquet"])
-                parquet_analyze = gr.Button("Parquet 파일 분석")
+            file_upload = gr.File(
+                label="파일 업로드 (.cod, .txt, .py, .parquet)",
+                file_types=[".cod", ".txt", ".py", ".parquet"],
+                type="file"
+            )
             
             with gr.Accordion("고급 설정", open=False):
                 system_message = gr.Textbox(label="System Message", value="")
@@ -116,38 +96,19 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", 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 analyze_file(file_type):
-        if file_type == "fashion":
-            return "패션 코드 실행"
-        elif file_type == "uhd":
-            return "UHD 이미지 코드 실행"
-        elif file_type == "mixgen":
-            return "MixGEN 코드 실행"
-        elif file_type == "parquet":
-            return "test.parquet 실행"
-
-    # 채팅 제출 핸들러
-    def chat(message, history):
-        return respond(
-            message=message,
-            history=history,
-            fashion_file=fashion_file.value,
-            uhd_file=uhd_file.value,
-            mixgen_file=mixgen_file.value,
-            parquet_file=parquet_file.value,
-            system_message=system_message.value,
-            max_tokens=max_tokens.value,
-            temperature=temperature.value,
-            top_p=top_p.value,
-        )
-
     # 이벤트 바인딩
-    msg.submit(chat, [msg, chatbot], [msg, chatbot])
-    fashion_analyze.click(lambda: analyze_file("fashion"), None, msg)
-    uhd_analyze.click(lambda: analyze_file("uhd"), None, msg)
-    mixgen_analyze.click(lambda: analyze_file("mixgen"), None, msg)
-    parquet_analyze.click(lambda: analyze_file("parquet"), None, msg)
+    msg.submit(
+        chat,
+        inputs=[msg, chatbot, file_upload, system_message, max_tokens, temperature, top_p],
+        outputs=[msg, chatbot]
+    )
+
+    # 파일 업로드 시 자동 분석
+    file_upload.change(
+        chat,
+        inputs=[gr.Textbox(value="파일 분석을 시작합니다."), chatbot, file_upload, system_message, max_tokens, temperature, top_p],
+        outputs=[msg, chatbot]
+    )
 
     # 예제 추가
     gr.Examples(