Update app.py

app.py

@@ -28,7 +28,7 @@ def create_article_components(max_results):
     return article_components
     
 API_KEY = os.getenv("SERPHOUSE_API_KEY")
-# hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
+hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
 
 # 국가별 언어 코드 매핑
 COUNTRY_LANGUAGES = {
@@ -40,6 +40,7 @@ COUNTRY_LANGUAGES = {
     "Germany": "de",
     "France": "fr",
     "Japan": "ja",
+#    "South Korea": "ko",
     "China": "zh",
     "India": "hi",
     "Brazil": "pt",
@@ -110,6 +111,7 @@ COUNTRY_LOCATIONS = {
     "Germany": "Germany",
     "France": "France",
     "Japan": "Japan",
+#    "South Korea": "South Korea",
     "China": "China",
     "India": "India",
     "Brazil": "Brazil",
@@ -308,6 +310,51 @@ def serphouse_search(query, country):
     response_data = search_serphouse(query, country)
     return format_results_from_raw(response_data)
 
+
+# Hacker News API 관련 함수 추가
+def get_hn_item(item_id):
+    """개별 아이템 정보 가져오기"""
+    try:
+        response = requests.get(f"https://hacker-news.firebaseio.com/v0/item/{item_id}.json")
+        return response.json()
+    except:
+        return None
+
+def get_recent_stories():
+    """최신 스토리 가져오기"""
+    try:
+        # newstories 엔드포인트로 최신 스토리 ID 목록 가져오기
+        response = requests.get("https://hacker-news.firebaseio.com/v0/newstories.json")
+        story_ids = response.json()
+        
+        # 24시간 이내 스토리만 필터링
+        recent_stories = []
+        current_time = datetime.now().timestamp()
+        day_ago = current_time - (24 * 60 * 60)
+        
+        for story_id in story_ids:
+            story = get_hn_item(story_id)
+            if story and 'time' in story and story['time'] > day_ago:
+                recent_stories.append(story)
+            
+            # 충분한 스토리를 모았으면 중단
+            if len(recent_stories) >= 100:  # 최대 100개로 제한
+                break
+                
+        return recent_stories
+    except Exception as e:
+        print(f"Error fetching HN stories: {str(e)}")
+        return []
+
+def format_hn_time(timestamp):
+    """Unix timestamp를 읽기 쉬운 형식으로 변환"""
+    try:
+        dt = datetime.fromtimestamp(timestamp)
+        return dt.strftime("%Y-%m-%d %H:%M:%S")
+    except:
+        return "Unknown time"
+
+        
 css = """
 footer {visibility: hidden;}
 
@@ -618,4 +665,80 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         outputs=global_search_outputs
     )
 
+# 탭 구현 부분에 다음 코드 추가
+        with gr.Tab("AI 리포터"):
+            gr.Markdown("지난 24시간 동안의 Hacker News 포스트를 보여줍니다.")
+    
+            with gr.Column():
+                refresh_button = gr.Button("새로고침", variant="primary")
+                status_message_hn = gr.Markdown("")
+        
+        # 결과 출력 영역
+                with gr.Column(elem_id="hn_results_area"):
+                    hn_articles_state = gr.State([])
+            
+                    hn_article_components = []
+                    for i in range(100):  # 최대 100개 표시
+                        with gr.Group(visible=False) as article_group:
+                            title = gr.Markdown()
+                            info = gr.Markdown()
+                    
+                            hn_article_components.append({
+                                'group': article_group,
+                                'title': title,
+                                'info': info,
+                                'index': i,
+                            })
+
+        def refresh_hn_stories():
+    """Hacker News 스토리 새로고침"""
+            status_msg = "Hacker News 포스트를 가져오는 중..."
+    
+            outputs = [gr.update(value=status_msg, visible=True)]
+    
+    # 모든 컴포넌트 초기화
+            for _ in hn_article_components:
+                outputs.extend([
+                    gr.update(visible=False),
+                    gr.update(),
+                    gr.update()
+                ])
+    
+    # 최신 스토리 가져오기
+            stories = get_recent_stories()
+    
+    # 결과 업데이트
+            outputs = [gr.update(value=f"총 {len(stories)}개의 포스트를 찾았습니다.", visible=True)]
+    
+            for idx, comp in enumerate(hn_article_components):
+                if idx < len(stories):
+                    story = stories[idx]
+                    outputs.extend([
+                        gr.update(visible=True),
+                        gr.update(value=f"### [{story.get('title', 'Untitled')}]({story.get('url', '#')})"),
+                        gr.update(value=f"**작성자:** {story.get('by', 'unknown')} | **시간:** {format_hn_time(story.get('time', 0))} | **점수:** {story.get('score', 0)} | **댓글:** {len(story.get('kids', []))}개")
+                    ])
+                else:
+                    outputs.extend([
+                        gr.update(visible=False),
+                        gr.update(),
+                        gr.update()
+                    ])
+    
+            return outputs
+
+# 이벤트 연결
+        hn_outputs = [status_message_hn]
+        for comp in hn_article_components:
+            hn_outputs.extend([
+                comp['group'],
+                comp['title'],
+                comp['info']
+            ])
+        
+        refresh_button.click(
+            refresh_hn_stories,
+            outputs=hn_outputs
+        )
+
 iface.launch(auth=("it1","chosun1"))