Update app.py

app.py

@@ -910,27 +910,33 @@ def respond(
     top_p,
 ):
     if not url.startswith('http'):
-        return "Please enter a valid URL"
-    
-    # 기사 내용 추출
-    article_content = get_article_content(url)
-    
-    # 번역 요청을 위한 프롬프트 구성
-    translation_prompt = f"""Please translate the following article to Korean. 
-    Maintain the original meaning and context while making it natural in Korean.
-    
-    Article: {article_content}
-    
-    Korean translation:"""
-    
-    messages = [
-        {"role": "system", "content": system_message or "You are a helpful translator that translates articles to Korean."},
-        {"role": "user", "content": translation_prompt}
-    ]
+        history.append((url, "올바른 URL을 입력해주세요."))
+        return history
 
-    response = ""
-    
     try:
+        # 기사 내용 추출
+        article_content = get_article_content(url)
+        
+        # 번역 요청을 위한 프롬프트 구성
+        translation_prompt = f"""Please translate the following article to Korean. 
+        Maintain the original meaning and context while making it natural in Korean.
+        
+        Article: {article_content}
+        
+        Korean translation:"""
+        
+        messages = [
+            {"role": "system", "content": system_message or "You are a helpful translator that translates articles to Korean."},
+            {"role": "user", "content": translation_prompt}
+        ]
+
+        # 사용자 입력을 히스토리에 추가
+        history.append((url, "번역을 시작합니다..."))
+        
+        # 번역 결과를 저장할 변수
+        full_response = ""
+        
+        # 스트리밍 응답 처리
         for message in client.chat.completions.create(
             model="CohereForAI/c4ai-command-r-plus-08-2024",
             max_tokens=max_tokens,
@@ -939,15 +945,23 @@ def respond(
             top_p=top_p,
             messages=messages,
         ):
-            token = message.choices[0].delta.content
-            if token:
-                response += token
-                yield response
+            if hasattr(message.choices[0].delta, 'content'):
+                token = message.choices[0].delta.content
+                if token:
+                    full_response += token
+                    # 마지막 메시지를 업데이트
+                    history[-1] = (url, full_response)
+                    yield history
+                    
     except Exception as e:
-        yield f"Translation error: {str(e)}"
+        error_message = f"번역 중 오류가 발생했습니다: {str(e)}"
+        history.append((url, error_message))
+        yield history
+
+    return history
 
 
-with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
+with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비스") as iface:
     with gr.Tabs():
         # 국가별 탭
         with gr.Tab("국가별"):
@@ -1065,7 +1079,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                         label="Top-P"
                     )
             
-                translate_button = gr.Button("번역", variant="primary")
+                translate_button = gr.Button("기사 생성", variant="primary")
             
             # 이벤트 연결
                 translate_button.click(