Update app.py

app.py

@@ -1,67 +1,61 @@
+import os
 import streamlit as st
+import openai
 import requests
-import httpx
 
-# Mediastack API configuration
+# Mediastack API 설정
 MEDIASTACK_API_KEY = "4b15f5b4bc2d12500c65817f202c500c"
 MEDIASTACK_URL = f"https://api.mediastack.com/v1/news?access_key={MEDIASTACK_API_KEY}&countries=kr"
 
-# HF Nexusflow API configuration
+# GLHF API 설정
 HF_API_KEY = "glhf_9ea0e0babe1e45353dd03b44cb979e22"
-HF_BASE_URL = "https://glhf.chat/api/openai/v1"
+openai.api_key = HF_API_KEY
+openai.api_base = "https://glhf.chat/api/openai/v1"
 
 def get_news():
+    """뉴스 가져오기"""
     response = requests.get(MEDIASTACK_URL)
     if response.status_code == 200:
         return response.json().get('data', [])
     return []
 
 def summarize_with_hf(text):
-    """요약 작업을 위해 HF Nexusflow 모델 호출"""
+    """뉴스 제목 요약"""
     try:
-        client = httpx.Client(follow_redirects=True, timeout=30.0)
-        payload = {
-            "model": "hf:Nexusflow/Athene-V2-Chat",
-            "messages": [
-                {"role": "system", "content": "뉴스 제목을 20자 이내의 한국어로 요약해 주세요.중국어는 절대 사용하지 말고, Note를 표시하지 마세요"},
+        completion = openai.ChatCompletion.create(
+            model="hf:Nexusflow/Athene-V2-Chat",
+            messages=[
+                {"role": "system", "content": "뉴스 제목을 20자 이내의 한국어로 요약해 주세요. 중국어는 절대 사용하지 말고, Note를 표시하지 마세요."},
                 {"role": "user", "content": text}
             ],
-            "temperature": 0.5
-        }
-        headers = {
-            "Authorization": f"Bearer {HF_API_KEY}",
-            "Content-Type": "application/json"
-        }
-        response = client.post(f"{HF_BASE_URL}/chat/completions", json=payload, headers=headers)
-        if response.status_code == 200:
-            return response.json()["choices"][0]["message"]["content"].strip()
+            temperature=0.5
+        )
+        return completion.choices[0].message['content'].strip()
     except Exception as e:
         print(f"Error summarizing: {e}")
-    return "요약 실패"
+        return "요약 실패"
 
 def display_news():
-    news_items = get_news()[:5]  # Limit to 5 news items
-
+    """뉴스 출력"""
+    news_items = get_news()[:5]  # 상위 5개의 뉴스만 표시
     if not news_items:
         st.warning("뉴스를 가져오는데 실패했습니다.")
         return
-        
+
     for item in news_items:
         original_title = item.get('title', '')
         if not original_title:
             continue
-        
+
         summarized_title = summarize_with_hf(original_title)
-        if summarized_title:
-            st.markdown(f'''
-                <div class="headline-container">
-                    <div class="headline-text">{summarized_title}</div>
-                </div>
-            ''', unsafe_allow_html=True)
+        st.markdown(f'''
+            <div class="headline-container">
+                <div class="headline-text">{summarized_title}</div>
+            </div>
+        ''', unsafe_allow_html=True)
 
 def main():
     st.set_page_config(layout="wide")
-    
     st.markdown("""
         <style>
         .headline-container {
@@ -83,53 +77,10 @@ def main():
             background-repeat: no-repeat;
             background-position: center;
         }
-        #clock-container {
-            position: fixed;
-            bottom: 0;
-            left: 0;
-            right: 0;
-            z-index: 1000;
-            width: 100%;
-            max-width: 1200px;
-            margin: 0 auto;
-            padding: 0 20px;
-            text-align: center;
-            height: 300px;
-            display: flex;
-            align-items: center;
-            justify-content: center;
-        }
-        #clock {
-            font-size: 15em;
-            font-weight: bold;
-            color: black;
-            line-height: 1.2;
-            white-space: nowrap;
-        }
         </style>
     """, unsafe_allow_html=True)
 
     display_news()
 
-    st.markdown("""
-        <div id="clock-container">
-            <span id="clock"></span>
-        </div>
-        <script>
-        function updateClock() {
-            const now = new Date();
-            const options = {
-                timeZone: 'Asia/Seoul',
-                hour12: true,
-                hour: 'numeric',
-                minute: '2-digit'
-            };
-            document.getElementById('clock').textContent = now.toLocaleTimeString('ko-KR', options);
-        }
-        setInterval(updateClock, 1000);
-        updateClock();
-        </script>
-    """, unsafe_allow_html=True)
-
 if __name__ == "__main__":
-    main()
+    main()