Update app.py

app.py

@@ -437,7 +437,7 @@ def run_ma240(ticker):
 # ============================================================
 # 3) 전량매도 (RS·이격)  (sell_+10_debug.py 로직 그대로)
 # ============================================================
-def run_sell(ticker):
+def run_sell(ticker, market_override="자동"):
     buf = io.StringIO()
     try:
         with contextlib.redirect_stdout(buf):
@@ -453,19 +453,35 @@ def run_sell(ticker):
             print(f"분석 종목: {TICKER}")
             print("=" * 70)
             print("\n[1/3] 시장 판별 중...")
-            market = 'KOSDAQ'; name = TICKER
-            try:
-                kospi  = fdr.StockListing('KOSPI')
-                kosdaq = fdr.StockListing('KOSDAQ')
-                if TICKER in kospi['Code'].values:
-                    market = 'KOSPI'
-                    name = kospi[kospi['Code'] == TICKER]['Name'].values[0]
-                elif TICKER in kosdaq['Code'].values:
+            market = None; name = TICKER
+            mo = (market_override or "자동").strip()
+            if mo in ('KOSPI', 'KOSDAQ'):
+                market = mo
+                print(f"    (수동 지정) {market}")
+            else:
+                try:
+                    for mk in ('KOSPI', 'KOSDAQ'):
+                        lst = fdr.StockListing(mk)
+                        code_col = next((c for c in lst.columns
+                                         if str(c).lower() in ('code', 'symbol', '종목코드')), None)
+                        name_col = next((c for c in lst.columns
+                                         if str(c).lower() in ('name', '종목명', 'korname')), None)
+                        codes = (lst[code_col] if code_col is not None else lst.index).astype(str)
+                        hit = (codes == TICKER)
+                        if bool(hit.any()):
+                            market = mk
+                            if name_col is not None:
+                                name = str(lst.loc[hit, name_col].values[0])
+                            break
+                    if market is None:
+                        market = 'KOSDAQ'
+                        print(f"    목록에서 못 찾음 -> KOSDAQ 가정 (위 '시장' 메뉴에서 직접 선택 가능)")
+                    else:
+                        print(f"    {name} ({TICKER}) / {market}")
+                except Exception as e:
                     market = 'KOSDAQ'
-                    name = kosdaq[kosdaq['Code'] == TICKER]['Name'].values[0]
-                print(f"    {name} ({TICKER}) / {market}")
-            except:
-                print(f"    판별 실패 -> {market} 가정")
+                    print(f"    판별 실패({type(e).__name__}) -> KOSDAQ 가정. "
+                          f"위 '시장' 메뉴에서 KOSPI/KOSDAQ를 직접 골라줘.")
 
             print("\n[2/3] 데이터 로드 중...")
             stock_df = fdr.DataReader(TICKER, DATA_START, ANALYZE_TO)
@@ -917,7 +933,7 @@ SELL       = "전량매도 (RS·이격)"
 CHOICES = [VCP_DETECT, VCP_SELL, BOLL, MA240, SELL]
 
 
-def analyze(choice, ticker, diag_date):
+def analyze(choice, ticker, diag_date, market_sel):
     ticker = (ticker or "").strip()
     if not ticker:
         return "티커 번호(6자리)를 입력하세요.\n예: 005930(삼성전자), 009150(삼성전기)"
@@ -932,7 +948,7 @@ def analyze(choice, ticker, diag_date):
     if choice == MA240:
         return run_ma240(ticker)
     if choice == SELL:
-        return run_sell(ticker)
+        return run_sell(ticker, market_sel)
     return "분석 종류를 선택하세요."
 
 
@@ -949,11 +965,15 @@ with gr.Blocks(title="주식 디버거", theme=gr.themes.Soft()) as demo:
         label="VCP 진단 날짜 (선택 · 'VCP 돌파 감지'에서만 사용)",
         placeholder="예: 2025-12-16  (비워두면 진단 생략)",
     )
+    market_sel = gr.Dropdown(
+        choices=["자동", "KOSPI", "KOSDAQ"], value="자동",
+        label="시장 (전량매도용 · 자동 판별 실패 시 직접 선택)",
+    )
     btn = gr.Button("분석 실행", variant="primary")
     out = gr.Code(label="결과", lines=30)
 
-    btn.click(analyze, inputs=[choice, ticker, diag_date], outputs=out)
-    ticker.submit(analyze, inputs=[choice, ticker, diag_date], outputs=out)
+    btn.click(analyze, inputs=[choice, ticker, diag_date, market_sel], outputs=out)
+    ticker.submit(analyze, inputs=[choice, ticker, diag_date, market_sel], outputs=out)
 
 
 if __name__ == "__main__":