Upload 4 files

app.py

@@ -1,147 +1,135 @@
-import gradio as gr
-import logging
 import math
-import FinanceDataReader as fdr
 import requests
+from bs4 import BeautifulSoup
+import FinanceDataReader as fdr
 import ssl
+import io
+import base64
+import gradio as gr
+import matplotlib.pyplot as plt
 from datetime import datetime
+from concurrent.futures import ThreadPoolExecutor
+import pytz
+import yfinance as yf
+from datetime import datetime, timedelta  # timedelta 추가
+import gradio as gr
+from gradio.components import Dataset
 import matplotlib.pyplot as plt
+import FinanceDataReader as fdr
+import gradio as gr
+import pandas as pd
+from concurrent.futures import ThreadPoolExecutor, as_completed
 import io
 import base64
-from concurrent.futures import ThreadPoolExecutor
-
-# 현재 날짜를 "Jun-20-2024" 형식으로 가져오기
-current_date = datetime.now().strftime("%b-%d-%Y")
-
-# SSL 인증서 검증 비활성화 (주석처리, 기본 HTTPS 컨텍스트 사용)
-# ssl._create_default_https_context = ssl._create_unverified_context
-
-# 로깅 설정
-logging.basicConfig(level=logging.WARNING)
-logger = logging.getLogger(__name__)
 
-exchange_rates = {}
 
-class StockCodeNotFoundError(Exception):
-    pass
 
-class CurrencyConversionError(Exception):
-    pass
+# 한국 표준시 (KST) 시간대 설정
+kst = pytz.timezone('Asia/Seoul')
 
-def parse_input(text, cash_amount_expr, cash_ratio):
-    logger.info(f"Parsing input: {text} with cash amount expression: {cash_amount_expr} and cash ratio: {cash_ratio}")
-    try:
-        lines = text.strip().split(',')
-        stock_inputs = []
-        total_target_weight = 0
-        
-        for line in lines:
-            parts = line.split()
-            if len(parts) == 4:
-                country_code, stock_code, quantity_expr, target_weight_expr = parts
-                quantity = math.floor(eval(quantity_expr.replace(' ', '')))
-                target_weight = eval(target_weight_expr.replace(' ', ''))
-                stock_inputs.append((country_code, stock_code, quantity, target_weight, target_weight_expr))  # target_weight_expr 추가
-                total_target_weight += target_weight
-            else:
-                raise ValueError("Invalid input format.")
-
-        if cash_amount_expr.strip():
-            cash_amount = math.floor(eval(cash_amount_expr.replace(' ', '')))
-        else:
-            cash_amount = 0
-
-        krw_cash = {'amount': cash_amount, 'target_weight': cash_ratio / 100.0}
+# SSL 인증서 검증 비활성화
+ssl._create_default_https_context = ssl._create_unverified_context
 
-        cash_ratio = krw_cash['target_weight']
-        stock_total_weight = total_target_weight
+def parse_input(text, cash_amount, cash_ratio):
+    lines = text.strip().split(',')
+    stock_inputs = []
+    total_target_weight = 0
+    
+    for line in lines:
+        parts = line.split()
+        if len(parts) == 4:
+            stock_code, currency_code, quantity_expr, target_weight_expr = parts
+            quantity = math.floor(eval(quantity_expr.replace(' ', '')))
+            target_weight = eval(target_weight_expr.replace(' ', ''))
+            target_ratio = (1 - cash_ratio / 100) * target_weight
+            stock_inputs.append((currency_code, stock_code, quantity, target_weight, target_ratio))
+            total_target_weight += target_weight
 
-        for i in range(len(stock_inputs)):
-            stock_inputs[i] = (stock_inputs[i][0], stock_inputs[i][1], stock_inputs[i][2], (1 - cash_ratio) * stock_inputs[i][3] / stock_total_weight, stock_inputs[i][4])
+    cash_amount = math.floor(cash_amount) if cash_amount else 0
+    krw_cash = {'amount': cash_amount, 'target_weight': cash_ratio / 100.0}
 
-        logger.info(f"Parsed stock inputs: {stock_inputs}, KRW cash: {krw_cash}")
-        return stock_inputs, krw_cash
-    except (ValueError, SyntaxError) as e:
-        logger.error("Error parsing input", exc_info=e)
-        raise ValueError("Invalid input format. Example: usd schd 21 8, krw 368590 530 2")
+    stock_total_weight = total_target_weight
 
-def get_exchange_rate(country_code):
-    if country_code.upper() in exchange_rates:
-        return exchange_rates[country_code.upper()]
+    for i in range(len(stock_inputs)):
+        stock_inputs[i] = (stock_inputs[i][0], stock_inputs[i][1], stock_inputs[i][2], stock_inputs[i][3], (1 - krw_cash['target_weight']) * stock_inputs[i][3] / stock_total_weight)
 
-    logger.info(f"Fetching exchange rate - {country_code}")
-    if country_code.upper() == 'KRW':
-        return 1
+    return stock_inputs, krw_cash
 
-    url = f"https://quotation-api-cdn.dunamu.com/v1/forex/recent?codes=FRX.KRW{country_code.upper()}"
-    try:
-        response = requests.get(url, verify=True)  # verify=True로 설정
-        response.raise_for_status()
-        data = response.json()
-        exchange_rate = data[0]['basePrice']
-        exchange_rates[country_code.upper()] = exchange_rate
-        logger.info(f"Exchange rate - {country_code}: {exchange_rate}")
-        return exchange_rate
-    except (requests.RequestException, IndexError) as e:
-        logger.error(f"Error fetching exchange rate - {country_code}", exc_info=e)
-        raise CurrencyConversionError("Error fetching exchange rate. Please enter a valid country code.")
-
-def get_exchange_reflected_stock_price(stock_code, country_code):
-    logger.info(f"Fetching exchange reflected stock price - {stock_code} in {country_code}")
-    try:
-        current_price = get_current_stock_price(stock_code)
-        exchange_rate = get_exchange_rate(country_code)
-        reflected_price = math.floor(current_price * exchange_rate)
-        logger.info(f"Reflected stock price - {stock_code}: {reflected_price}")
-        return reflected_price
-    except (StockCodeNotFoundError, CurrencyConversionError) as e:
-        logger.error(f"Error fetching reflected stock price - {stock_code} in {country_code}", exc_info=e)
-        raise e
+def get_exchange_rate(currency_code):
+    if currency_code.lower() == 'krw':
+        return 1.0
+    
+    ticker = f"{currency_code.upper()}KRW=X"
+    data = yf.download(ticker, period='1d')
+    if not data.empty:
+        return data['Close'].iloc[0]
+    else:
+        raise ValueError("Failed to retrieve exchange rate data.")
+
+def get_exchange_reflected_stock_price(stock_code, currency_code):
+    new_price = get_current_stock_price(stock_code)
+    exchange_rate = get_exchange_rate(currency_code)
+    return math.floor(new_price * exchange_rate)
 
 def get_current_stock_price(stock_code):
-    logger.info(f"Fetching current stock price - {stock_code}")
-    try:
-        df = fdr.DataReader(stock_code)
-        current_price = df['Close'].iloc[-1]
-        logger.info(f"Current stock price - {stock_code}: {current_price}")
-        return current_price
-    except ValueError as e:
-        logger.error(f"Error fetching stock price - {stock_code}", exc_info=e)
-        raise StockCodeNotFoundError("Stock code not found. Please enter a valid stock code.")
+    df = fdr.DataReader(stock_code)
+    return df['Close'].iloc[-1]
 
 def build_portfolio(stock_inputs, krw_cash):
     portfolio = {}
     target_weights = {}
-
-    logger.info(f"Building portfolio - stock inputs: {stock_inputs}, KRW cash: {krw_cash}")
-
+    
     with ThreadPoolExecutor() as executor:
-        results = list(executor.map(lambda stock_input: (
-            stock_input[1],  # stock_code
-            get_exchange_reflected_stock_price(stock_input[1], stock_input[0]),
-            stock_input[2],  # quantity
-            stock_input[3],  # target_weight
-            stock_input[4]   # target_weight_expr
-        ), stock_inputs))
-
-    for stock_code, current_price, quantity, target_weight, target_weight_expr in results:
-        portfolio[stock_code] = {'quantity': quantity, 'price': current_price, 'target_weight_expr': target_weight_expr}
-        target_weights[stock_code] = target_weight
-
-    if krw_cash is None:
-        krw_cash = {'amount': 0, 'target_weight': 0}
-
-    logger.info(f"Portfolio built: {portfolio}, target weights: {target_weights}, KRW cash: {krw_cash}")
+        results = executor.map(lambda x: (x[1], get_exchange_reflected_stock_price(x[1], x[0]), x[2], x[3], x[4], x[0]), stock_inputs)
+    
+    for stock_code, new_price, quantity, target_weight, target_ratio, currency_code in results:
+        portfolio[stock_code] = {'quantity': quantity, 'price': new_price, 'target_weight': target_weight, 'currency': currency_code}
+        target_weights[stock_code] = target_ratio
+
     return portfolio, target_weights, krw_cash
 
+def format_quantity(quantity):
+    if quantity < 0:
+        return f"({-quantity:,})"
+    else:
+        return f"{quantity:,}"
+
 def get_portfolio_rebalancing_info(portfolio, target_weights, krw_cash):
-    logger.info("Calculating portfolio rebalancing information")
+    with open('portfolio.html', 'r', encoding='utf-8') as file:
+        css = file.read()
+
+    kst = pytz.timezone('Asia/Seoul')
+    current_time = datetime.now(kst).strftime("%I:%M %p %b-%d-%Y")
 
     total_value = sum(stock['price'] * stock['quantity'] for stock in portfolio.values()) + krw_cash['amount']
     total_new_stock_value = 0
     total_trade_value = 0
     adjustments = []
 
+    # Calculate current weights and values
+    current_weights = {stock_code: (stock['price'] * stock['quantity'] / total_value) * 100 for stock_code, stock in portfolio.items()}
+    current_values = {stock_code: stock['price'] * stock['quantity'] for stock_code, stock in portfolio.items()}
+    
+    # Include cash in current weights and values
+    current_weights['CASH'] = (krw_cash['amount'] / total_value) * 100
+    current_values['CASH'] = krw_cash['amount']
+
+    # Sort stocks by current weight in descending order
+    sorted_stocks = sorted(current_weights.items(), key=lambda x: x[1], reverse=True)
+
+    # Display current weights and values section
+    current_info_html = "<h3>Your Portfolio Holdings</h3><div class='table-container'><table style='border-collapse: collapse;'>"
+    current_info_html += "<thead><tr><th style='border: 1px hidden #ddd; text-align: center;'>Stock Code</th><th style='border: 1px hidden #ddd; text-align: center;'>Current Weight (%)</th><th style='border: 1px hidden #ddd; text-align: center;'>Current Value</th></tr></thead><tbody>"
+    for stock_code, weight in sorted_stocks:
+        current_info_html += (
+            f"<tr>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{stock_code.upper()}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{weight:.1f}%</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>₩{current_values[stock_code]:,.0f}</td>"
+            f"</tr>"
+        )
+    current_info_html += "</tbody></table></div><br>"
+
     for stock_code, stock_data in portfolio.items():
         current_value = stock_data['price'] * stock_data['quantity']
         target_value = total_value * target_weights.get(stock_code, 0)
@@ -153,103 +141,400 @@ def get_portfolio_rebalancing_info(portfolio, target_weights, krw_cash):
         total_trade_value += abs(trade_value)
         total_new_stock_value += new_value
         current_value_pct = (current_value / total_value) * 100
+        new_value_pct = (new_value / total_value) * 100
 
-        adjustments.append((difference, current_value, target_value, current_value_pct, trade_quantity, stock_code, stock_data['price'], new_value, trade_value, stock_data['quantity'], new_quantity, stock_data['target_weight_expr']))  # target_weight_expr 추가
+        adjustments.append((difference, current_value, target_value, current_value_pct, trade_quantity, stock_code, stock_data['price'], new_value, trade_value, stock_data['quantity'], new_quantity, target_weights[stock_code], new_value_pct, stock_data['target_weight'], stock_data['currency']))
 
-    # 주식 평가금액 기준으로 정렬
-    adjustments.sort(key=lambda x: x[1], reverse=True)
+    krw_new_amount = total_value - total_new_stock_value
+    krw_target_value = total_value * krw_cash['target_weight']
+    krw_difference = krw_new_amount - krw_cash['amount']
+    trade_quantity = krw_difference
+    new_quantity = krw_cash['amount'] + trade_quantity
+    new_value = new_quantity
+    trade_value = trade_quantity
+    current_value = krw_cash['amount']
+    current_value_pct = (current_value / total_value) * 100
+    new_value_pct = (new_value / total_value) * 100
 
-    result_message = "<ol>"
-    for adjustment in adjustments:
-        difference, current_value, target_value, current_value_pct, trade_quantity, stock_code, price, new_value, trade_value, old_quantity, new_quantity, target_weight_expr = adjustment  # target_weight_expr 추가
-        new_value_pct = (new_value / total_value) * 100
-        formatted_trade_quantity = f"<span style='color: {'#57bb8a' if trade_quantity > 0 else '#cc0000'}'> {trade_quantity:+,} </span>" if trade_quantity != 0 else "0"
-        result_message += (
-            f"<li style='font-size: 1rem;'><strong>{stock_code.upper()}</strong> {current_value_pct:.0f}% <span style='background-color: yellow; color: blue;'>[ {target_weight_expr} ]</span> {target_value / total_value * 100:.0f}% {formatted_trade_quantity}</li>"  # target_weight_expr로 변경
+    adjustments.append((krw_difference, current_value, krw_target_value, current_value_pct, trade_quantity, 'CASH', 1, new_value, trade_value, krw_cash['amount'], new_quantity, krw_cash['target_weight'], new_value_pct, '', 'KRW'))
+
+    portfolio_info = css + f"""
+        <div><br>
+            <p><span style='font-size: 1.6rem; font-weight: bold;'>₩{total_value:,.0f}</span> as of <span style='color: #6e6e73;'>{current_time}</span></p>
+        <br></div>
+        """
+
+    currency_totals = {stock_data['currency']: {'amount': 0, 'weight': 0} for stock_data in portfolio.values()}
+
+    for stock_code, stock_data in portfolio.items():
+        currency = stock_data['currency']
+        current_value = stock_data['price'] * stock_data['quantity']
+        currency_totals[currency]['amount'] += current_value
+        currency_totals[currency]['weight'] += current_value / total_value
+
+    currency_totals['CASH'] = {'amount': krw_cash['amount'], 'weight': krw_cash['amount'] / total_value}
+    sorted_currencies = sorted(currency_totals.items(), key=lambda x: x[1]['weight'], reverse=True)
+
+    currency_table = "<h3>Your Portfolio by Currency</h3><div class='table-container'><table style='border-collapse: collapse;'>"
+    currency_table += "<thead><tr><th style='border: 1px hidden #ddd; text-align: center;'>Currency</th><th style='border: 1px hidden #ddd; text-align: center;'>Total Weight (%)</th><th style='border: 1px hidden #ddd; text-align: center;'>Total Value</th></tr></thead><tbody>"
+
+    for currency, data in sorted_currencies:
+        currency_table += (
+            f"<tr>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{currency.upper()}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{data['weight'] * 100:.1f}%</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>₩{data['amount']:,}</td>"            
+            f"</tr>"
+        )
+
+    currency_table += "</tbody></table></div><br>"
+
+    result_message = portfolio_info + current_info_html + currency_table + "<h3>Re-Balancing Analysis</h3><div class='table-container'><table style='border-collapse: collapse;'>"
+    result_message += "<thead><tr><th style='border: 1px hidden #ddd; text-align: center;'>Stock Code</th><th style='border: 1px hidden #ddd; text-align: center;'>Target Weight</th><th style='border: 1px hidden #ddd; text-align: center;'>Target Ratio (%)</th><th style='border: 1px hidden #ddd; text-align: center;'>Buy or Sell?</th><th style='border: 1px hidden #ddd; text-align: center;'>Trade Amount</th><th style='border: 1px hidden #ddd; text-align: center;'>Current Price per Share</th><th style='border: 1px hidden #ddd; text-align: center;'>Estimated # of<br> Shares to Buy or Sell</th><th style='border: 1px hidden #ddd; text-align: center;'>Quantity of Units</th><th style='border: 1px hidden #ddd; text-align: center;'>Market Value</th><th style='border: 1px hidden #ddd; text-align: center;'>% Asset Allocation</th></tr></thead><tbody>"
+
+    for adj in adjustments:
+        difference, current_value, target_value, current_value_pct, trade_quantity, stock_code, price, new_value, trade_value, old_quantity, new_quantity, target_ratio, new_value_pct, target_weight, currency = adj
+        Buy_or_Sell = ""
+        if trade_quantity > 0:
+            Buy_or_Sell = f"<span class='buy-sell buy'>Buy</span>"
+        elif trade_quantity < 0:
+            Buy_or_Sell = f"<span class='buy-sell sell'>Sell</span>"
+        else:
+            Buy_or_Sell = f"<span></span>"
+
+        price_str = f"₩{price:,.0f}" if stock_code != 'CASH' else ''
+        target_weight_str = f"<span class='highlight-edit'>{target_weight}</span>" if stock_code != 'CASH' else ''
+        target_ratio_str = f"<span class='highlight-edit'>{target_ratio * 100:.1f}%</span>" if stock_code == 'CASH' else f"{target_ratio * 100:.1f}%"
+        old_quantity_str = f"{old_quantity:,.0f} → {new_quantity:,.0f}" if stock_code != 'CASH' else ''
+        trade_value_str = f"<span class='highlight-sky'>{format_quantity(trade_value)}</span>" if trade_value != 0 else ''
+        trade_quantity_str = (
+            f"<span class='highlight-sky'>{format_quantity(trade_quantity)}</span>"
+            if stock_code != 'CASH' and trade_value != 0 else ''
         )
-    result_message += "</ol>"
-
-    if krw_cash:
-        krw_new_amount = total_value - total_new_stock_value
-        krw_target_value = total_value * krw_cash['target_weight']
-        krw_difference = krw_new_amount - krw_cash['amount']
-        krw_new_pct = (krw_new_amount / total_value) * 100
-        formatted_krw_difference = f"<span style='color: {'#57bb8a' if krw_difference > 0 else '#cc0000'}'> {krw_difference:+,} </span>" if krw_difference != 0 else "0"
+        new_value_str = f"₩{new_value:,.0f}"
+        new_value_pct_str = f"{new_value_pct:.1f}%"
+
         result_message += (
-            f"<p style='font-size: 1rem;'><strong>💰 CASH</strong> {krw_cash['amount'] / total_value * 100:.0f}% <span style='background-color: yellow; color: blue;'>[ {krw_cash['target_weight'] * 100:.0f}% ]</span> {formatted_krw_difference}</p>"
+            f"<tr>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{stock_code.upper()}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{target_weight_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{target_ratio_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{Buy_or_Sell}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{trade_value_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{price_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{trade_quantity_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{old_quantity_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{new_value_str}</td>"
+            f"<td style='border: 1px hidden #ddd; text-align: center;'>{new_value_pct_str}</td>"
+            f"</tr>"
         )
 
-    result_message += f"<p style='font-size: 1.5rem;'><strong>🦄 {total_value:,}</strong></p>"
-    logger.info("Portfolio rebalancing information calculated")
-    return result_message, adjustments, total_value, krw_new_amount
-
-def plot_bar_chart(adjustments):
-    fig, ax = plt.subplots()
-    stock_codes = [f"{adj[5].upper()}\n@{adj[6]:,}" for adj in adjustments]
-    current_values = [adj[1] for adj in adjustments]
-    new_values = [adj[7] for adj in adjustments]
-
-    x = range(len(stock_codes))
-    bar_width = 0.35
-    bars1 = ax.bar([i - bar_width / 2 for i in x], current_values, width=bar_width, label='CURRENT VALUE')
-    bars2 = ax.bar([i + bar_width / 2 for i in x], new_values, width=bar_width, label='NEW VALUE')
-
-    ax.set_ylabel('VALUE (IN KRW)')
-    ax.set_xticks(x)
-    ax.set_xticklabels(stock_codes)
-    ax.legend()
-
-    for bar in bars1:
-        yval = bar.get_height()
-        ax.text(bar.get_x() + bar.get_width() / 2, yval, f"{int(yval):,}", ha='center', va='bottom')
-
-    for bar in bars2:
-        yval = bar.get_height()
-        ax.text(bar.get_x() + bar.get_width() / 2, yval, f"{int(yval):,}", ha='center', va='bottom')
-    
-    plt.tight_layout()  # 그래프와 타이틀 사이의 간격 조정
+    result_message += "</tbody></table></div>"
     
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
-    img = base64.b64encode(buf.read()).decode('utf-8')
-    buf.close()
+    return result_message
 
-    return f'<img src="data:image/png;base64,{img}"/>'
-
-def rebalance_portfolio(input_text, cash_amount_expr, cash_ratio):
+def rebalancing_tool(user_input, cash_amount, cash_ratio):
     try:
-        # 입력 데이터 파싱
-        stock_inputs, krw_cash = parse_input(input_text, cash_amount_expr, cash_ratio)
-        
-        # 포트폴리오 구축
+        stock_inputs, krw_cash = parse_input(user_input, cash_amount, cash_ratio)
         portfolio, target_weights, krw_cash = build_portfolio(stock_inputs, krw_cash)
-        
-        # 포트폴리오 리밸런싱 정보 계산
-        result_message, adjustments, total_value, krw_new_amount = get_portfolio_rebalancing_info(portfolio, target_weights, krw_cash)
-        
-        # 현재 vs 새로운 값 바 차트 생성
-        bar_chart = plot_bar_chart(adjustments)
-        
-        # 결과 반환
-        return result_message, bar_chart
+        result = get_portfolio_rebalancing_info(portfolio, target_weights, krw_cash)
+        return result
+    except Exception as e:
+        return str(e)
+
+def get_stock_prices(stock_code, days):
+    try:
+        df = fdr.DataReader(stock_code, end=pd.Timestamp.now().date(), data_source='yahoo')
+        df = df[df.index >= df.index.max() - pd.DateOffset(days=days)]  # 최근 days일 데이터로 제한
+        return df['Close']
     except Exception as e:
-        return "", str(e)
-
-# Gradio 인터페이스 설정
-interface = gr.Interface(
-    fn=rebalance_portfolio,
-    inputs=[
-        gr.Textbox(lines=2, value="krw 458730 530 8, krw 368590 79 2", label="포트폴리오 데이터"),
-        gr.Textbox(lines=1, value="518192", label="추가 투자금"),
-        gr.Slider(minimum=0, maximum=100, step=1, value=0, label="현금 비율 (%)")
+        print(f"Failed to fetch data for {stock_code}: {e}")
+        return None
+
+def plot_stock_prices(stock_codes, days):
+    # 주식 그래프 생성을 위한 병렬 처리
+    stock_prices = {}
+    with ThreadPoolExecutor() as executor:
+        futures = {executor.submit(get_stock_prices, stock_code.strip(), int(days)): stock_code.strip() for stock_code in stock_codes.split(',')}
+        for future in as_completed(futures):
+            stock_code = futures[future]
+            try:
+                prices = future.result()
+                if prices is not None:
+                    stock_prices[stock_code] = prices
+            except Exception as e:
+                print(f"Failed to fetch data for {stock_code}: {e}")
+
+    # 각 주식에 대한 그래프를 그림
+    plt.figure(figsize=(10, 6))
+    for stock_code, prices in stock_prices.items():
+        relative_prices = prices / prices.iloc[0]  # 첫 번째 데이터 포인트를 기준으로 상대적 가격 계산
+        plt.plot(prices.index, relative_prices, label=stock_code.upper())  # 주식 코드를 대문자로 표시
+    plt.xlabel('Date')
+    plt.ylabel('Relative Price (Normalized to 1)')
+    plt.title(f'Relative Stock Prices Over the Last {days} Days')
+    plt.legend()
+
+    # 그래프를 HTML로 변환하여 반환
+    html_graph = io.BytesIO()
+    plt.savefig(html_graph, format='png', dpi=300)
+    html_graph.seek(0)
+    graph_encoded = base64.b64encode(html_graph.getvalue()).decode()
+    graph_html = f'<img src="data:image/png;base64,{graph_encoded}"/>'
+
+    return graph_html
+
+def cost_averaging(old_avg_price, old_quantity, new_price, new_quantity):
+    # 입력값을 숫자로 변환
+    old_avg_price = float(old_avg_price) if old_avg_price else 0.0
+    old_quantity = float(old_quantity) if old_quantity else 0.0
+    new_price = float(new_price) if new_price else 0.0
+    new_quantity = float(new_quantity) if new_quantity else 0.0
+
+    # 현재 투자 금액 계산
+    current_investment = old_avg_price * old_quantity
+    # 추가 투자 금액 계산
+    additional_investment = new_price * new_quantity
+    # 총 투자 금액
+    total_investment = current_investment + additional_investment
+    # 총 주식 수
+    total_shares = old_quantity + new_quantity
+    # 새 평균 가격 계산
+    new_avg_price = total_investment / total_shares if total_shares != 0 else 0.0
+    
+    # 현재 수익률 계산
+    current_return = (new_price - old_avg_price) / old_avg_price * 100 if old_avg_price != 0 else 0.0
+    # 새로운 수익률 계산
+    new_return = (new_price / new_avg_price - 1 ) * 100 if new_avg_price != 0 else 0.0
+    return new_avg_price, total_shares, total_investment, current_return, new_return, additional_investment
+
+def gradio_cost_averaging(old_avg_price, old_quantity, new_price, new_quantity):
+    with open('portfolio.html', 'r', encoding='utf-8') as file:
+        css = file.read()
+
+    # 입력값을 숫자로 변환
+    old_avg_price = float(old_avg_price) if old_avg_price else 0.0
+    old_quantity = float(old_quantity) if old_quantity else 0.0
+    new_price = float(new_price) if new_price else 0.0
+    new_quantity = float(new_quantity) if new_quantity else 0.0
+
+    new_avg_price, total_shares, total_investment, current_return, new_return, additional_investment = cost_averaging(old_avg_price, old_quantity, new_price, new_quantity)
+    
+    current_return_class = ""
+    if current_return > 0:
+        current_return_class = f"<span style='color: #4caf50; font-weight: bold;'>{current_return:+,.2f}%</span>"
+    elif current_return < 0:
+        current_return_class = f"<span style='color: #f44336; font-weight: bold;'>{current_return:,.2f}%</span>"
+    else:
+        current_return_class = f"<span><strong>0</strong></span>"
+
+    new_return_class = ""
+    if new_return > 0:
+        new_return_class = f"<span style='color: #4caf50; font-weight: bold;'>{new_return:+,.2f}%</span>"
+    elif current_return < 0:
+        new_return_class = f"<span style='color: #f44336; font-weight: bold;'>{new_return:,.2f}%</span>"
+    else:
+        new_return_class = f"<span><strong>0</strong></span>"
+
+    # Construct the HTML string with the appropriate class
+    result_html = css+ f"""
+<div style="box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1); border-radius: 8px; padding: 48px; position: relative; width: 100%; padding: 24px;">
+    <div>
+        <div style="margin-bottom: 24px;">
+            <div style="font-size: 24px; margin-bottom: 24px;">Average Price</div>
+            <div style="font-size: 24px; font-weight: bold; color: #1c75bc;">
+                <span></span>
+                <span>{new_avg_price:,.0f}</span>
+            </div>
+            <hr style="margin: 24px 0;">
+        </div>
+    </div>
+    <div>
+        <div style="margin-bottom: 24px;">
+            <div style="font-size: 24px; margin-bottom: 24px;">Total Quantity</div>
+            <div style="font-size: 24px; font-weight: bold; color: #1c75bc;">
+                <span>{total_shares:,.0f}</span>
+            </div>
+            <hr style="margin: 24px 0;">
+        </div>
+    </div>
+    <div>
+        <div style="margin-bottom: 24px;">
+            <div style="font-size: 24px; margin-bottom: 24px;">Total Investment</div>
+            <div style="font-size: 24px; font-weight: bold; color: #1c75bc;">
+                <span></span>
+                <span>{total_investment:,.0f}</span>
+            </div>
+            <hr style="margin: 24px 0;">
+        </div>
+    </div>
+    <div style='display: flex; justify-content: space-around; align-items: center;'>
+        <div style='text-align: center;'>
+            <div style="margin-bottom: 24px;">
+                <div style="font-size: 24px; margin-bottom: 24px;"></div>
+                <div style="font-size: 24px; font-weight: bold; color: #1c75bc;">
+                    <p></p>
+                    <p>{current_return_class}</p>
+                    <p>{old_avg_price:,.0f}</p>
+                </div>
+            </div>
+        </div>
+        <div style='text-align: center; margin-bottom: 24px; font-size: 24px;'>➜</div>
+        <div style='text-align: center;'>
+            <div style="margin-bottom: 24px;">
+                <div style="font-size: 24px; margin-bottom: 24px;"></div>
+                <div style="font-size: 24px; font-weight: bold; color: #1c75bc;">
+                    <p></p>
+                    <p>{new_return_class}</p>
+                    <p>{new_avg_price:,.0f}</p>
+                </div>
+            </div>
+        </div>
+    </div>
+    <p style='text-align: center;'>총 추가 금액: <strong>{additional_investment:,.0f}</strong></p>
+</div>
+
+    """
+
+    return result_html
+
+
+
+
+
+# Define the interface for the Portfolio tab
+def portfolio_interface(input_text, cash_amount, cash_ratio):
+    result = rebalancing_tool(input_text, cash_amount, cash_ratio)
+    return result
+
+portfolio_inputs = [
+    gr.Textbox(label="🔥 Holdings", lines=2, placeholder="Format: [stock code  currency  quantity  target weight,    ...]"),
+    gr.Number(label="🪵 Cash", value=""),
+    gr.Slider(label="⚖️ Cash Ratio (%)", minimum=0, maximum=100, step=1)
+]
+
+portfolio_interface = gr.Interface(
+    fn=portfolio_interface,
+    inputs=portfolio_inputs,
+    outputs=gr.HTML(),
+    examples = [
+    ["458730 krw 571 8,\n368590 krw 80 2", 17172, 0],
+    ["SCHD USD 400 8,\nQQQ USD 40 2", 1000000, 25],
+    ["458730 krw 571 8,\n368590 krw 80 2,\nSCHD USD 400 8,\nQQQ USD 40 2", 1000000, 25]
     ],
-    outputs=[
-        gr.HTML(label="결과 메시지"),
-        gr.HTML(label="바 차트")
+    live=False
+)
+
+# Define the interface for the Compare tab
+def compare_interface(stock_codes, period):
+    result = plot_stock_prices(stock_codes, period)
+    return result
+
+compare_inputs = [
+    gr.Textbox(label="📈 Stock Codes", lines=2, placeholder="Enter stock codes separated by comma (e.g., AAPL,GOOGL,MSFT)"),
+    gr.Number(label="📆 Period (days)", value=90)
+]
+
+compare_interface = gr.Interface(
+    fn=compare_interface,
+    inputs=compare_inputs,
+    outputs=gr.HTML(),
+    examples = [
+    ["SCHD,QQQ", 90],
+    ["458730,368590", 90],
+    ["AAPL,GOOGL,MSFT", 90]
+    ],
+    live=False
+)
+
+# Define the interface for the Cost Averaging tab
+def cost_averaging_interface(old_avg_price, old_quantity, new_price, new_quantity):
+    result = gradio_cost_averaging(old_avg_price, old_quantity, new_price, new_quantity)
+    return result
+
+cost_averaging_inputs = [
+    gr.Number(label="Old Price", value=""),
+    gr.Number(label="Quantity", value=""),
+    gr.Number(label="New Price", value=""),
+    gr.Number(label="Quantity", value="")
+]
+
+cost_averaging_interface = gr.Interface(
+    fn=cost_averaging_interface,
+    inputs=cost_averaging_inputs,
+    outputs=gr.HTML(),
+    examples = [
+    [78.15, 6.024272, 77.11, 1]
     ],
-    title=f"🦊🐻🐶🐰{current_date}"
+    live=False
 )
 
-if __name__ == "__main__":
-    interface.launch(share=True)
+# Combine all interfaces into a tabbed interface
+with gr.Blocks(css='style.css') as demo:
+    with gr.Column(elem_id="col-container"):
+        with gr.Tabs():
+            with gr.TabItem("Portfolio"):
+                portfolio_interface.render()
+            with gr.TabItem("Compare"):
+                compare_interface.render()
+            with gr.TabItem("Cost Averaging"):
+                cost_averaging_interface.render()
+            with gr.TabItem("📄 About"):
+                gr.Markdown("""
+                # About This Application
+
+                Welcome to the Portfolio Management Tool! This application provides a comprehensive suite of tools to help you manage and analyze your investment portfolio. Below is a brief overview of each feature available in this tool.
+
+                ## 📊 Portfolio
+
+                **Description:**  
+                This section allows you to analyze and rebalance your investment portfolio. You can input your current holdings, cash amount, and desired cash ratio, and the tool will calculate the necessary trades to achieve your target allocation.
+
+                **How to Use:**  
+                1. Enter your holdings in the format: `[stock code currency quantity target weight]`.
+                2. Specify your cash amount and desired cash ratio.
+                3. Click the "Analyze Data" button to see the rebalancing analysis.
+                4. View the detailed breakdown of your current portfolio and suggested trades.
+
+                ## 📈 Compare
+
+                **Description:**  
+                This feature enables you to compare the historical prices of multiple stocks over a specified period. It provides a visual comparison to help you understand the performance of different stocks.
+
+                **How to Use:**  
+                1. Enter the stock codes separated by commas (e.g., AAPL, GOOGL, MSFT).
+                2. Specify the period in days for which you want to compare the stock prices.
+                3. Click the "Compare Stock Prices" button to generate the comparison graph.
+                4. View the relative price changes of the selected stocks over the chosen period.
+
+                ## 💹 Cost Averaging
+
+                **Description:**  
+                This section helps you calculate the new average price of a stock when you make additional purchases. It also provides insights into the current and new return rates based on your investments.
+
+                **How to Use:**  
+                1. Enter the average price and quantity of your initial purchase in the "First Purchase" section.
+                2. Enter the price and quantity of your subsequent purchase in the "Second Purchase" section.
+                3. Click the "Calculate Cost Averaging" button to see the results.
+                4. View the new average price, total quantity, total investment, and return rates.
+
+                ## 📄 About
+
+                **Description:**  
+                This section provides an overview of the application, explaining its features and how to use them. It serves as a guide for new users to understand the functionalities available in the tool.
+
+                **How to Use:**  
+                Simply read through the information provided to get acquainted with the application's capabilities.
+
+                ## Disclaimer
+
+                Please note that this tool is for informational purposes only and does not constitute financial advice. Always conduct your own research or consult with a financial advisor before making investment decisions.
+
+                ---
+
+                We hope you find this tool useful for managing your investments. If you have any feedback or suggestions, feel free to reach out!
+
+                Happy Investing!
+                """)
+
+demo.launch(share=True)

portfolio.html

@@ -0,0 +1,249 @@
+<!DOCTYPE html>
+<html lang="ko">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Portfolio Rebalancing</title>
+    <style>
+        :root {
+            --background-color-light: #ffffff;
+            --background-color-dark: #121212;
+            --text-color-light: #333333;
+            --text-color-dark: #ffffff;
+            --highlight-color-light: #f7f7f7;
+            --highlight-color-dark: #1e1e1e;
+            --header-color-light: #000000;
+            --header-color-dark: #ffffff;
+            --buy-color: #4caf50;
+            --sell-color: #f44336;
+            --highlight-edit-bg-color-light: #fff2cc;
+            --highlight-edit-text-color-light: #0000ff;
+            --highlight-edit-bg-color-dark: hsl(45, 100%, 70%); /* 어두운 배경 */
+            --highlight-edit-text-color-dark: hsl(240, 100%, 50%); /* 밝은 텍스트 */
+            --highlight-yellow-bg-color-light: #ffeb3b;
+            --highlight-yellow-bg-color-dark: #ffca28;
+            --highlight-yellow-text-color-light: #000000;
+            --highlight-yellow-text-color-dark: #000000;
+            --highlight-sky-bg-color-light: #ddf5fd;
+            --highlight-sky-bg-color-dark: #89c9e6;
+            --highlight-sky-text-color-light: #000000;
+            --highlight-sky-text-color-dark: #000000;
+            --highlight-black-light: #000000;
+            --highlight-black-dark: #ffffff;
+            --total-value-color-light: #000000;
+            --total-value-color-dark: #ffeb3b;
+        }
+
+        body {
+            font-family: 'Roboto', sans-serif;
+            line-height: 1.6;
+            color: var(--text-color-light);
+            background-color: var(--background-color-light);
+            padding: 20px;
+        }
+
+        .buy-sell {
+            padding: 5px 10px;
+            border-radius: 5px;
+            font-weight: bold;
+            display: inline-block;
+            margin: 2px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        .buy {
+            background-color: var(--buy-color);
+            color: white !important;
+        }
+
+        .dashboard {
+            /* background-color: #f6fcfe; */
+            /* border: 1px solid #89c9e6; */
+            text-align: left;
+            padding: 10px;
+            font-size: 1rem;
+            border-radius: 18px;
+            box-shadow: 2px 4px 12px #00000014;
+            transition: all .3s cubic-bezier(0,0,.5,1);
+        }
+
+
+        .sell {
+            background-color: var(--sell-color);
+            color: white !important;
+        }
+        .highlight-edit {
+            background-color: var(--highlight-edit-bg-color-light);
+            color: var(--highlight-edit-text-color-light) !important;
+            padding: 5px 10px;
+            font-weight: bold;
+            border-radius: 5px;
+            display: inline-block;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+        .highlight-black {
+            background-color: var(--highlight-black-light);
+            color: var(--text-color-dark) !important;
+            padding: 5px 10px;
+            font-weight: bold;
+            border-radius: 5px;
+            display: inline-block;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        .highlight-yellow {
+            background-color: var(--highlight-yellow-bg-color-light);
+            color: var(--highlight-yellow-text-color-light) !important;
+            padding: 5px 10px;
+            font-weight: bold;
+            border-radius: 5px;
+            display: inline-block;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        .highlight-sky {
+            background-color: var(--highlight-sky-bg-color-light);
+            color: var(--highlight-sky-text-color-light) !important;
+            padding: 5px 10px;
+            font-weight: bold;
+            border-radius: 5px;
+            display: inline-block;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        .container {
+            font-size: 1.2rem;
+            margin-bottom: 15px;
+            padding: 20px;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+            background-color: var(--highlight-color-light);
+        }
+
+        .header {
+            font-size: 1.2rem;
+            font-weight: bold;
+            color: var(--header-color-light);
+            margin-top: 10px;
+            text-align: center;
+        }
+
+        .total-value {
+            color: var(--total-value-color-light);
+            font-size: 3rem;
+        }
+
+        .summary {
+            font-size: 1.2rem;
+            color: var(--text-color-light);
+            margin-top: 10px;
+            text-align: center;
+        }
+
+        .table-container {
+            width: 100%;
+            overflow: auto;
+            margin-bottom: 20px;
+            position: relative;
+            max-height: 600px;
+            border: 1px hidden #ddd;
+            max-height: max-content; /* 표의 최대 높이 설정 */
+            overflow-y: hidden; /* 스크롤 숨김 */
+        }
+
+        .table-container table {
+            width: 100%;
+            border-collapse: collapse;
+        }
+
+        .table-container th, .table-container td {
+            border: 1px hidden #ddd;
+            padding: 8px;
+            text-align: left;
+            background-color: var(--background-color-light);
+        }
+
+        .table-container th {
+            background-color: var(--highlight-color-light);
+            position: sticky;
+            top: 0;
+            z-index: 2;
+        }
+
+        .table-container td:first-child, .table-container th:first-child {
+            position: sticky;
+            left: 0;
+            z-index: 1;
+        }
+
+        .table-container th:first-child {
+            z-index: 3;
+        }
+
+        @media (prefers-color-scheme: dark) {
+            body {
+                color: var(--text-color-dark);
+                background-color: var(--background-color-dark);
+            }
+
+            .container {
+                background-color: var(--highlight-color-dark);
+            }
+
+            .header {
+                color: var(--header-color-dark);
+            }
+
+            .total-value {
+                color: var(--total-value-color-dark);
+            }
+
+            .table-container th, .table-container td {
+                background-color: var(--background-color-dark);
+                color: var(--text-color-dark);
+            }
+
+            .highlight-yellow {
+                background-color: var(--highlight-yellow-bg-color-dark);
+                color: var(--highlight-yellow-text-color-dark) !important;
+            }
+
+            .highlight-black {
+                background-color: var(--highlight-black-dark);
+                color: var(--text-color-light) !important;
+            }
+
+            .highlight-sky {
+                background-color: var(--highlight-sky-bg-color-dark);
+                color: var(--highlight-sky-text-color-dark) !important;
+            }
+            .highlight-edit {
+                background-color: var(--highlight-edit-bg-color-dark);
+                color: var(--highlight-edit-text-color-dark) !important;
+            }
+
+            .table-container th {
+                background-color: var(--highlight-color-dark);
+            }
+
+            .buy-sell, .highlight-black, .highlight-yellow, .highlight-sky, .highlight-edit {
+                box-shadow: 0 2px 4px rgba(255, 255, 255, 0.1);
+            }
+            .dashboard {
+                background-color: #2b2b2b; /* 어두운 배경색 */
+                border: 1px solid #555555; /* 어두운 테두리 색상 */
+                color: #e0e0e0; /* 밝은 글자 색상 */
+                text-align: left;
+                padding: 10px;
+                font-size: 1rem;
+                border-radius: 18px;
+                box-shadow: 2px 4px 12px #00000050; /* 다크모드에 어울리는 그림자 색상 */
+                transition: all .3s cubic-bezier(0,0,.5,1);
+            }
+
+
+        }
+    </style>
+</head>
+</html>

requirements.txt

@@ -4,3 +4,7 @@ finance-datareader
 requests
 plotly
 gradio
+matplotlib
+pandas
+pytz
+yfinance

style.css

@@ -0,0 +1,13 @@
+
+@import url('https://fonts.googleapis.com/css2?family=Montserrat:wght@400;500;700&display=swap');
+@import url('https://fonts.googleapis.com/css2?family=IBM+Plex+Mono:wght@400;500;700&display=swap');
+
+#col-container {
+    margin: 0 auto;
+    max-width: 100%;
+    font-family: 'Montserrat', 'ui-sans-serif', 'system-ui', 'sans-serif';
+}
+
+.code {
+    font-family: 'IBM Plex Mono', 'ui-monospace', 'Consolas', 'monospace';
+}