Update main.py

main.py

@@ -253,6 +253,17 @@ def groq_use(text_content: Any) -> str:
         logger.error(f"🚨 Groq 호출 실패: {repr(e)}")
         return "추출 실패"
     
+def get_stock_info(company_info: str) -> Dict[str, str] | None:
+    try:
+        if krx_listings is not None:
+            # 포함 관계 확인
+            for _, row in krx_listings.iterrows():
+                if row['Name'] in company_info:
+                    return {"market": "KRX", "symbol": row['Code'], "name": row['Name']}
+    except Exception as e:
+        logger.error(f"주식 검색 에러: {e}")
+    return None
+
 
 # ---------------------------------------
 # 1) 요약 단계
@@ -538,35 +549,27 @@ async def produce_progress(analysis_id: str, user_id: str | None, percent: int,
 
 def get_stock_info(company_name: str) -> Dict[str, str] | None:
     try:
-        name = (company_name or "").strip()
-        if not name:
-            return None
-
-        # 1) KRX
-        if krx_listings is not None and not krx_listings.empty:
-            # code 컬럼 자동 선택
-            code_col = 'Code' if 'Code' in krx_listings.columns else ('Symbol' if 'Symbol' in krx_listings.columns else None)
-            if code_col and 'Name' in krx_listings.columns:
-                kr_match = krx_listings[krx_listings['Name'].str.contains(name, case=False, na=False)]
-                if not kr_match.empty:
-                    s = kr_match.iloc[0]
-                    code = str(s[code_col]).zfill(6)
-                    return {"market": "KRX", "symbol": code, "name": s['Name']}
-
-        # 2) US
-        if us_listings is not None and not us_listings.empty:
-            if 'Name' in us_listings.columns and 'Symbol' in us_listings.columns:
-                us_match = us_listings[
-                    us_listings['Name'].str.contains(name, case=False, na=False) |
-                    us_listings['Symbol'].str.fullmatch(name, case=False)
-                ]
-                if not us_match.empty:
-                    s = us_match.iloc[0]
-                    return {"market": "US", "symbol": str(s['Symbol']), "name": s['Name']}
+        # 1. 한국 주식(KRX) 검색
+        if krx_listings is not None:
+            kr_match = krx_listings[krx_listings['Name'].str.contains(company_name, case=False, na=False)]
+            if not kr_match.empty:
+                s = kr_match.iloc[0]
+                return {"market": "KRX", "symbol": s['Code'], "name": s['Name']}
+
+        # 2. 미국 주식 검색 (번역기 없이 이름으로 직접 검색)
+        if us_listings is not None:
+            # 영어 이름이 포함되어 있을 수 있으므로 대소문자 무시하고 검색
+            us_match = us_listings[
+                us_listings['Name'].str.contains(company_name, case=False, na=False) |
+                us_listings['Symbol'].str.fullmatch(company_name, case=False)
+            ]
+            if not us_match.empty:
+                s = us_match.iloc[0]
+                return {"market": "US", "symbol": s['Symbol'], "name": s['Name']}
 
     except Exception as e:
-        logger.error(f"주식 종목 검색 중 오류 발생: {repr(e)}", exc_info=True)
-
+        logger.error(f"주식 종목 검색 중 오류 발생: {e}")
+    
     return None
     
 
@@ -640,12 +643,13 @@ async def consume_loop():
             # ✅ threadpool에서 호출될 progress 콜백
             def progress_cb(percent: int, stage: str, message: str):
                 # threadpool(다른 스레드) -> 현재 이벤트루프에서 코루틴 실행
-                try:
-                    asyncio.run_coroutine_threadsafe(
+                fut = asyncio.run_coroutine_threadsafe(
                     produce_progress(analysis_id, user_id, percent, stage, message),
                     loop
-                    )
-                
+                )
+                # (선택) progress 전송 실패 로그 보고 싶으면:
+                try:
+                    fut.result(timeout=2)
                 except Exception as e:
                     logger.warning(f"[KAFKA] progress send failed: {e}")
 
@@ -665,9 +669,7 @@ async def consume_loop():
                         "userId": user_id,
                         "result": result
                     }, ensure_ascii=False).encode("utf-8")
-                    
-                    logger.info("DONE topic=%s bytes=%d", KAFKA_DONE_TOPIC, len(done_payload))
-                    
+
                     await producer.send_and_wait(
                         KAFKA_DONE_TOPIC,
                         key=analysis_id.encode("utf-8"),
@@ -678,22 +680,6 @@ async def consume_loop():
             except Exception as e:
                 logger.error(f"[KAFKA] analysis failed analysisId={analysis_id}: {repr(e)}", exc_info=True)
                 await produce_progress(analysis_id, user_id, 100, "ERROR", f"오류: {type(e).__name__}")
-                if producer:
-                    err_payload = {
-                    "analysisId": analysis_id,
-                    "userId": user_id,
-                    "result": {
-                    "error": True,
-                    "message": f"분석 실패: {type(e).__name__}",
-                    "url": url,
-                    }
-                    }
-                    await producer.send_and_wait(
-                        KAFKA_DONE_TOPIC,
-                        key=analysis_id.encode("utf-8"),
-                        value=json.dumps(err_payload, ensure_ascii=False).encode("utf-8")
-                    )
-                    logger.info(f"[KAFKA] produced error-done analysisId={analysis_id}")
 
     finally:
         logger.info("[KAFKA] stopped")