Update app-backup2.py

app-backup2.py

@@ -10,7 +10,74 @@ from requests.packages.urllib3.util.retry import Retry
 from openai import OpenAI
 from bs4 import BeautifulSoup
 import re  # re 모듈 추가
+import json
+import os
+from datetime import datetime
+import sqlite3
+import pathlib
+
+# DB 초기화 함수
+def init_db():
+    db_path = pathlib.Path("search_results.db")
+    conn = sqlite3.connect(db_path)
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS searches
+                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
+                  keyword TEXT,
+                  country TEXT,
+                  results TEXT,
+                  timestamp DATETIME DEFAULT CURRENT_TIMESTAMP)''')
+    conn.commit()
+    conn.close()
+
+# 검색 결과 저장 함수
+def save_to_db(keyword, country, results):
+    conn = sqlite3.connect("search_results.db")
+    c = conn.cursor()
+    c.execute("INSERT INTO searches (keyword, country, results) VALUES (?, ?, ?)",
+              (keyword, country, json.dumps(results)))
+    conn.commit()
+    conn.close()
+
+# DB에서 검색 결과 불러오기 함수
+def load_from_db(keyword, country):
+    conn = sqlite3.connect("search_results.db")
+    c = conn.cursor()
+    c.execute("SELECT results, timestamp FROM searches WHERE keyword=? AND country=? ORDER BY timestamp DESC LIMIT 1",
+              (keyword, country))
+    result = c.fetchone()
+    conn.close()
+    if result:
+        return json.loads(result[0]), result[1]
+    return None, None
+
+# 삼성/미국 검색 함수
+def search_samsung_us():
+    error_message, articles = serphouse_search("samsung", "United States")
+    if not error_message and articles:
+        save_to_db("samsung", "United States", articles)
+        return display_results(articles)
+    return "검색 결과가 없습니다."
 
+# DB에서 삼성/미국 결과 불러오기 함수
+def load_samsung_us():
+    results, timestamp = load_from_db("samsung", "United States")
+    if results:
+        return f"저장 시간: {timestamp}\n\n" + display_results(results)
+    return "저장된 결과가 없습니다."
+
+# 결과 표시 함수
+def display_results(articles):
+    output = ""
+    for idx, article in enumerate(articles, 1):
+        output += f"### {idx}. {article['title']}\n"
+        output += f"출처: {article['channel']}\n"
+        output += f"시간: {article['time']}\n"
+        output += f"링크: {article['link']}\n"
+        output += f"요약: {article['snippet']}\n\n"
+    return output
+
+    
 ACCESS_TOKEN = os.getenv("HF_TOKEN")
 if not ACCESS_TOKEN:
     raise ValueError("HF_TOKEN environment variable is not set")
@@ -725,6 +792,7 @@ def search_global(query, region, articles_state_global):
 css = """
 /* 전역 스타일 */
 footer {visibility: hidden;}
+
 /* 레이아웃 컨테이너 */
 #status_area {
     background: rgba(255, 255, 255, 0.9);
@@ -733,33 +801,40 @@ footer {visibility: hidden;}
     margin-bottom: 20px;
     box-shadow: 0 2px 5px rgba(0,0,0,0.1);
 }
+
 #results_area {
     padding: 10px;
     margin-top: 10px;
 }
+
 /* 탭 스타일 */
 .tabs {
     border-bottom: 2px solid #ddd !important;
     margin-bottom: 20px !important;
 }
+
 .tab-nav {
     border-bottom: none !important;
     margin-bottom: 0 !important;
 }
+
 .tab-nav button {
     font-weight: bold !important;
     padding: 10px 20px !important;
 }
+
 .tab-nav button.selected {
     border-bottom: 2px solid #1f77b4 !important;
     color: #1f77b4 !important;
 }
+
 /* 상태 메시지 */
 #status_area .markdown-text {
     font-size: 1.1em;
     color: #2c3e50;
     padding: 10px 0;
 }
+
 /* 기본 컨테이너 */
 .group {
     border: 1px solid #eee;
@@ -768,16 +843,19 @@ footer {visibility: hidden;}
     border-radius: 5px;
     background: white;
 }
+
 /* 버튼 스타일 */
 .primary-btn {
     background: #1f77b4 !important;
     border: none !important;
 }
+
 /* 입력 필드 */
 .textbox {
     border: 1px solid #ddd !important;
     border-radius: 4px !important;
 }
+
 /* 프로그레스바 컨테이너 */
 .progress-container {
     position: fixed;
@@ -788,6 +866,7 @@ footer {visibility: hidden;}
     background: #e0e0e0;
     z-index: 1000;
 }
+
 /* 프로그레스바 */
 .progress-bar {
     height: 100%;
@@ -796,6 +875,7 @@ footer {visibility: hidden;}
     transition: width 0.3s ease;
     animation: progress-glow 1.5s ease-in-out infinite;
 }
+
 /* 프로그레스 텍스트 */
 .progress-text {
     position: fixed;
@@ -810,6 +890,7 @@ footer {visibility: hidden;}
     z-index: 1001;
     box-shadow: 0 2px 5px rgba(0,0,0,0.2);
 }
+
 /* 프로그레스바 애니메이션 */
 @keyframes progress-glow {
     0% {
@@ -822,6 +903,7 @@ footer {visibility: hidden;}
         box-shadow: 0 0 5px rgba(33, 150, 243, 0.5);
     }
 }
+
 /* 반응형 디자인 */
 @media (max-width: 768px) {
     .group {
@@ -834,27 +916,32 @@ footer {visibility: hidden;}
         padding: 3px 10px;
     }
 }
+
 /* 로딩 상태 표시 개선 */
 .loading {
     opacity: 0.7;
     pointer-events: none;
     transition: opacity 0.3s ease;
 }
+
 /* 결과 컨테이너 애니메이션 */
 .group {
     transition: all 0.3s ease;
     opacity: 0;
     transform: translateY(20px);
 }
+
 .group.visible {
     opacity: 1;
     transform: translateY(0);
 }
+
 /* Examples 스타일링 */
 .examples-table {
     margin-top: 10px !important;
     margin-bottom: 20px !important;
 }
+
 .examples-table button {
     background-color: #f0f0f0 !important;
     border: 1px solid #ddd !important;
@@ -863,11 +950,13 @@ footer {visibility: hidden;}
     margin: 2px !important;
     transition: all 0.3s ease !important;
 }
+
 .examples-table button:hover {
     background-color: #e0e0e0 !important;
     transform: translateY(-1px) !important;
     box-shadow: 0 2px 5px rgba(0,0,0,0.1) !important;
 }
+
 .examples-table .label {
     font-weight: bold !important;
     color: #444 !important;
@@ -946,17 +1035,21 @@ def respond(url, history, system_message, max_tokens, temperature, top_p):
         article_content = get_article_content(url)
         
         translation_prompt = f"""다음 영문 기사를 한국어로 번역하고 기사를 작성해주세요.
+
 1단계: 전문 번역
 ===번역 시작===
 {article_content}
 ===번역 끝===
+
 2단계: 기사 작성 가이드라인
 다음 요구사항에 따라 한국어 기사를 작성하세요:
+
 1. 구조
 - 헤드라인: 핵심 내용을 담은 강력한 제목
 - 부제목: 헤드라인 보완 설명
 - 리드문: 기사의 핵심을 요약한 첫 문단
 - 본문: 상세 내용 전개
+
 2. 작성 규칙
 - 객관적이고 정확한 사실 전달
 - 문장은 '다.'로 종결
@@ -964,10 +1057,12 @@ def respond(url, history, system_message, max_tokens, temperature, top_p):
 - 인용구는 따옴표 처리
 - 핵심 정보를 앞부분에 배치
 - 전문 용어는 적절한 설명 추가
+
 3. 형식
 - 적절한 단락 구분
 - 읽기 쉬운 문장 길이
 - 논리적인 정보 구성
+
 각 단계는 '===번역===', '===기사==='로 명확히 구분하여 출력하세요.
 """
 
@@ -1004,9 +1099,81 @@ def respond(url, history, system_message, max_tokens, temperature, top_p):
 
     return history
 
+
+def continue_writing(history, system_message, max_tokens, temperature, top_p):
+    if not history:
+        return history
+    
+    last_response = history[-1][1] if history else ""
+    continue_prompt = f"""이전 내용을 이어서 계속 작성해주세요. 
+    마지막 응답: {last_response}
+    
+    추가 지침:
+    1. 이전 내용의 맥락을 유지하며 자연스럽게 이어서 작성
+    2. 새로운 정보나 상세 설명을 추가
+    3. 필요한 경우 보충 설명이나 분석 제공
+    4. 기사 형식과 스타일 유지
+    5. 필요한 경우 추가적인 이미지 프롬프트 생성
+    """
+
+    # 메시지 구조 수정
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": continue_prompt}  # 사용자 메시지로 시작
+    ]
+
+    try:
+        full_response = ""
+        for message in client.chat.completions.create(
+            model="CohereForAI/c4ai-command-r-plus-08-2024",
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+            messages=messages,
+        ):
+            if hasattr(message.choices[0].delta, 'content'):
+                token = message.choices[0].delta.content
+                if token:
+                    full_response += token
+                    # 이전 대화 기록을 유지하면서 새로운 응답 추가
+                    new_history = history.copy()
+                    new_history.append(("계속 작성", full_response))
+                    yield new_history
+
+    except Exception as e:
+        error_message = f"계속 작성 중 오류가 발생했습니다: {str(e)}"
+        new_history = history.copy()
+        new_history.append(("오류", error_message))
+        yield new_history
+
+    return history
+
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비스") as iface:
+    init_db()  # DB 초기화
+    
     with gr.Tabs():
-        # 국가별 탭
+        # DB 저장/불러오기 탭
+        with gr.Tab("DB 검색"):
+            gr.Markdown("삼성/미국 검색 결과를 DB에 저장하고 불러옵니다.")
+            
+            with gr.Row():
+                search_button = gr.Button("검색: samsung/미국", variant="primary")
+                load_button = gr.Button("출력: samsung/미국", variant="secondary")
+            
+            results_display = gr.Markdown()
+            
+            # 버튼 이벤트 연결
+            search_button.click(
+                fn=search_samsung_us,
+                outputs=results_display
+            )
+            
+            load_button.click(
+                fn=load_samsung_us,
+                outputs=results_display
+            )
+
         with gr.Tab("국가별"):
             gr.Markdown("검색어를 입력하고 원하는 국가(한국 제외)를를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 100개 출력합니다.")
             gr.Markdown("국가 선택후 검색어에 '한글'을 입력하면 현지 언어로 번역되어 검색합니다. 예: 'Taiwan' 국가 선택후 '삼성' 입력시 '三星'으로 자동 검색")
@@ -1049,7 +1216,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                 translated_query_display = gr.Markdown(visible=False)
                 search_button = gr.Button("검색", variant="primary")
             
-                
                 progress = gr.Progress()
                 articles_state = gr.State([])
 
@@ -1107,8 +1273,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                                 'index': i,
                             })
 
-    
-    # AI 번역 탭 추가
+        # AI 번역 탭
         with gr.Tab("AI 기사 생성"):
             gr.Markdown("뉴스 URL을 입력하면 AI가 한국어로 번역하여 기사 형식으로 작성합니다.")
             gr.Markdown("이미지 생성: https://huggingface.co/spaces/ginipick/FLUXllama ")
@@ -1122,10 +1287,13 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                         placeholder="https://..."
                     )
 
-                with gr.Accordion("고급 설정", open=False):
+                with gr.Row():
+                    translate_button = gr.Button("기사 생성", variant="primary")
+                    continue_button = gr.Button("계속 이어서 작성", variant="secondary")
 
+                with gr.Accordion("고급 설정", open=False):
                     system_message = gr.Textbox(
-                         value="""You are a professional translator and journalist. Follow these steps strictly:
+                        value="""You are a professional translator and journalist. Follow these steps strictly:
                          1. TRANSLATION
                          - Start with ===번역=== marker
                          - Provide accurate Korean translation
@@ -1155,14 +1323,12 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                          - Clearly separate each section with markers
                          - Never skip or combine steps
                          - Ensure image prompts align with article content""",
-                         label="System message"
-                     )                    
+                        label="System message"
+                    )
                     
-                
-
                     max_tokens = gr.Slider(
                         minimum=1,
-                    maximum=7800,
+                        maximum=7800,
                         value=7624,
                         step=1,
                         label="Max new tokens"
@@ -1181,26 +1347,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                         step=0.05,
                         label="Top-P"
                     )
-            
-                translate_button = gr.Button("기사 생성", variant="primary")
-            
-            # 이벤트 연결
-                translate_button.click(
-                    fn=respond,
-                    inputs=[
-                        url_input,
-                        chatbot,
-                        system_message,
-                        max_tokens,
-                        temperature,
-                        top_p,
-                    ],
-                    outputs=chatbot
-                )
-
-
 
-    
     # 이벤트 연결 부분
     # 국가별 탭 이벤트
     search_outputs = [status_message, translated_query_display, gr.Markdown(visible=False)]
@@ -1234,6 +1381,33 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
         show_progress=True
     )
 
+    # AI 번역 탭 이벤트
+    translate_button.click(
+        fn=respond,
+        inputs=[
+            url_input,
+            chatbot,
+            system_message,
+            max_tokens,
+            temperature,
+            top_p,
+        ],
+        outputs=chatbot
+    )
+
+    # 계속 작성 버튼 이벤트
+    continue_button.click(
+        fn=continue_writing,
+        inputs=[
+            chatbot,
+            system_message,
+            max_tokens,
+            temperature,
+            top_p,
+        ],
+        outputs=chatbot
+    )
+
 iface.launch(
     server_name="0.0.0.0",
     server_port=7860,
@@ -1241,4 +1415,4 @@ iface.launch(
     auth=("gini","pick"),
     ssl_verify=False,
     show_error=True
-)                    
+)