Update app.py

app.py

@@ -1,300 +1,162 @@
 import gradio as gr
-import yfinance as yf
 import pandas as pd
-import numpy as np
-import torch
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-from datetime import datetime, timedelta
-import plotly.graph_objects as go
-import plotly.express as px
-from plotly.subplots import make_subplots
-import warnings
-warnings.filterwarnings('ignore')
-import spaces
-
-# Import utility functions
+import yfinance as yf
+from datetime import datetime
 from utils import (
-    get_indonesian_stocks, 
+    get_indonesian_stocks,
     calculate_technical_indicators,
     generate_trading_signals,
     get_fundamental_data,
-    format_large_number,
     predict_prices,
     create_price_chart,
     create_technical_chart,
-    create_prediction_chart
+    create_prediction_chart,
 )
-from config import IDX_STOCKS, TECHNICAL_INDICATORS, PREDICTION_CONFIG
-
-# Load Chronos-Bolt model
-@spaces.GPU(duration=120)
-def load_model():
-    """Load the Amazon Chronos-Bolt model for time series forecasting"""
-    model = AutoModelForSeq2SeqLM.from_pretrained(
-        "amazon/chronos-bolt-base",
-        torch_dtype=torch.bfloat16,
-        device_map="auto",
-        trust_remote_code=True
-    )
-    tokenizer = None 
-    return model, tokenizer
+import warnings
 
-# Initialize model
-model, tokenizer = load_model()
+warnings.filterwarnings("ignore")
 
-def get_stock_data(symbol, period="1y"):
-    """Fetch historical stock data using yfinance"""
+def analyze_stock(symbol, prediction_days=30):
     try:
         stock = yf.Ticker(symbol)
-        data = stock.history(period=period)
+        data = stock.history(period="6mo", interval="1d")
         if data.empty:
-            return None, None
-        return data, stock
-    except Exception as e:
-        print(f"Error fetching data for {symbol}: {e}")
-        return None, None
-
-def analyze_stock(symbol, prediction_days=30):
-    """Main analysis function"""
-    # FIX: Add .JK suffix if it's missing (case-insensitive check)
-    if not symbol.upper().endswith(".JK"):
-        symbol += ".JK"
+            raise ValueError("No price data available.")
         
-    # Get stock data
-    data, stock = get_stock_data(symbol)
-    
-    if data is None or stock is None:
-        return None, None, None, None, None, None
-    
-    # Get fundamental data
-    fundamental_info = get_fundamental_data(stock)
-    
-    # Calculate technical indicators
-    indicators = calculate_technical_indicators(data)
-    
-    # Generate trading signals
-    signals = generate_trading_signals(data, indicators)
-    
-    # Make predictions using Chronos-Bolt
-    predictions = predict_prices(data, model, tokenizer, prediction_days)
-    
-    # Create charts
-    fig_price = create_price_chart(data, indicators) 
-    fig_technical = create_technical_chart(data, indicators) 
-    fig_prediction = create_prediction_chart(data, predictions) 
-    
-    return fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction
+        indicators = calculate_technical_indicators(data)
+        signals = generate_trading_signals(data, indicators)
+        fundamental_info = get_fundamental_data(stock)
+        predictions = predict_prices(data, prediction_days=prediction_days)
+        fig_price = create_price_chart(data, indicators)
+        fig_technical = create_technical_chart(data, indicators)
+        fig_prediction = create_prediction_chart(data, predictions)
 
-def create_ui():
-    """Create the Gradio interface"""
-    with gr.Blocks(
-        title="IDX Stock Analysis & Prediction",
-        theme=gr.themes.Soft(),
-        css="""
-        .header { 
-            text-align: center; 
-            padding: 20px; 
-            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-            color: white;
-            border-radius: 10px;
-            margin-bottom: 20px;
-        }
-        .metric-card {
-            background: white;
-            padding: 15px;
-            border-radius: 8px;
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-            margin: 10px 0;
+        return fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction, predictions
+    except Exception as e:
+        print(f"Error analyzing {symbol}: {e}")
+        empty_fig = gr.Plot.update(value=None)
+        empty_predictions = {
+            'high_30d': 0, 'low_30d': 0, 'change_pct': 0, 'summary': 'Prediction unavailable.'
         }
-        .positive { color: #10b981; font-weight: bold; }
-        .negative { color: #ef4444; font-weight: bold; }
-        .neutral { color: #6b7280; font-weight: bold; }
-        """
-    ) as demo:
-        
-        with gr.Row():
-            gr.HTML("""
-            <div class="header">
-                <h1>IDX Stock Analysis & Prediction</h1>
-                <p>Advanced Technical Analysis & AI-Powered Predictions for Indonesian Stock Exchange</p>
-            </div>
-            """)
-        
-        with gr.Row():
-            with gr.Column(scale=2):
-                # FIX: Change Dropdown to Textbox for flexible user input
-                stock_selector = gr.Textbox(
-                    value="BBCA",
-                    label="Input Simbol Saham IDX",
-                    info="Contoh: BBCA, ADRO, TLKM"
-                )
-                
-                with gr.Row():
-                    prediction_days = gr.Slider(
-                        minimum=7,
-                        maximum=90,
-                        value=30,
-                        step=7,
-                        label="Prediction Days"
-                    )
-                    analyze_btn = gr.Button(
-                        "Analyze Stock",
-                        variant="primary",
-                        size="lg"
-                    )
-        
-        # Results sections
-        with gr.Tabs() as tabs:
-            
-            # Tab 1: Stock Overview & Fundamentals
-            with gr.TabItem("Stock Overview"):
-                with gr.Row():
-                    company_name = gr.Textbox(label="Company Name", interactive=False)
-                    current_price = gr.Number(label="Current Price (IDR)", interactive=False)
-                    market_cap = gr.Textbox(label="Market Cap", interactive=False)
-                
-                with gr.Row():
-                    pe_ratio = gr.Number(label="P/E Ratio", interactive=False)
-                    dividend_yield = gr.Number(label="Dividend Yield (%)", interactive=False)
-                    volume = gr.Number(label="Volume", interactive=False)
-                
-                fundamentals_text = gr.Textbox(
-                    label="Company Information",
-                    lines=8,
-                    interactive=False
-                )
-            
-            # Tab 2: Technical Analysis
-            with gr.TabItem("Technical Analysis"):
-                price_chart = gr.Plot(label="Price & Technical Indicators")
-                technical_chart = gr.Plot(label="Technical Indicators Analysis")
-                
-                with gr.Row():
-                    rsi_value = gr.Number(label="RSI (14)", interactive=False)
-                    macd_signal = gr.Textbox(label="MACD Signal", interactive=False)
-                    bb_position = gr.Textbox(label="Bollinger Band Position", interactive=False)
-            
-            # Tab 3: Trading Signals
-            with gr.TabItem("Trading Signals"):
-                with gr.Row():
-                    overall_signal = gr.Textbox(label="Overall Signal", interactive=False, scale=2)
-                    signal_strength = gr.Slider(
-                        minimum=0,
-                        maximum=100,
-                        label="Signal Strength",
-                        interactive=False
-                    )
-                
-                signals_text = gr.Textbox(
-                    label="Detailed Signals",
-                    lines=10,
-                    interactive=False
-                )
-                
-                with gr.Row():
-                    support_level = gr.Number(label="Support Level", interactive=False)
-                    resistance_level = gr.Number(label="Resistance Level", interactive=False)
-                    stop_loss = gr.Number(label="Recommended Stop Loss", interactive=False)
-            
-            # Tab 4: AI Predictions
-            with gr.TabItem("AI Predictions"):
-                prediction_chart = gr.Plot(label="Price Forecast (Chronos-Bolt)")
-                
-                with gr.Row():
-                    predicted_high = gr.Number(label="Predicted High (30d)", interactive=False)
-                    predicted_low = gr.Number(label="Predicted Low (30d)", interactive=False)
-                    predicted_change = gr.Number(label="Expected Change (%)", interactive=False)
-                
-                prediction_summary = gr.Textbox(
-                    label="Prediction Analysis",
-                    lines=5,
-                    interactive=False
-                )
-        
-        # Event handlers
-        def update_analysis(symbol, pred_days):
-            fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction = analyze_stock(symbol, pred_days)
-            
-            if fundamental_info is None:
-                # FIX: Return appropriate initial values or error message
-                return {
-                    company_name: "Gagal memuat data",
-                    current_price: 0,
-                    market_cap: "N/A",
-                    pe_ratio: 0,
-                    dividend_yield: 0,
-                    volume: 0,
-                    fundamentals_text: f"Tidak dapat mengambil data saham untuk {symbol}. Pastikan simbol benar.",
-                    rsi_value: 0,
-                    macd_signal: "N/A",
-                    bb_position: "N/A",
-                    overall_signal: "N/A",
-                    signal_strength: 0,
-                    signals_text: "Tidak ada sinyal tersedia",
-                    support_level: 0,
-                    resistance_level: 0,
-                    stop_loss: 0,
-                    predicted_high: 0,
-                    predicted_low: 0,
-                    predicted_change: 0,
-                    prediction_summary: "Prediksi tidak tersedia",
-                    price_chart: go.Figure(),
-                    technical_chart: go.Figure(),
-                    prediction_chart: go.Figure()
-                }
-            
-            # Format outputs
-            return {
-                company_name: fundamental_info.get('name', 'N/A'),
-                current_price: fundamental_info.get('current_price', 0),
-                market_cap: format_large_number(fundamental_info.get('market_cap', 0)),
-                pe_ratio: fundamental_info.get('pe_ratio', 0),
-                dividend_yield: fundamental_info.get('dividend_yield', 0),
-                volume: fundamental_info.get('volume', 0),
-                fundamentals_text: fundamental_info.get('info', ''),
-                rsi_value: indicators.get('rsi', {}).get('current', 0),
-                macd_signal: indicators.get('macd', {}).get('signal', 'N/A'),
-                bb_position: indicators.get('bollinger', {}).get('position', 'N/A'),
-                overall_signal: signals.get('overall', 'HOLD'),
-                signal_strength: signals.get('strength', 50),
-                signals_text: signals.get('details', ''),
-                support_level: signals.get('support', 0),
-                resistance_level: signals.get('resistance', 0),
-                stop_loss: signals.get('stop_loss', 0),
-                predicted_high: indicators.get('predictions', {}).get('high_30d', 0),
-                predicted_low: indicators.get('predictions', {}).get('low_30d', 0),
-                predicted_change: indicators.get('predictions', {}).get('change_pct', 0),
-                prediction_summary: indicators.get('predictions', {}).get('summary', ''),
-                price_chart: fig_price,
-                technical_chart: fig_technical,
-                prediction_chart: fig_prediction
-            }
-        
-        analyze_btn.click(
-            fn=update_analysis,
-            inputs=[stock_selector, prediction_days],
-            outputs=[
-                company_name, current_price, market_cap, pe_ratio, dividend_yield, volume, fundamentals_text,
-                rsi_value, macd_signal, bb_position, overall_signal, signal_strength, signals_text,
-                support_level, resistance_level, stop_loss, predicted_high, predicted_low, predicted_change,
-                prediction_summary, price_chart, technical_chart, prediction_chart
-            ]
+        return {}, {}, {}, empty_fig, empty_fig, empty_fig, empty_predictions
+
+def update_analysis(symbol, prediction_days):
+    (
+        fundamental_info,
+        indicators,
+        signals,
+        fig_price,
+        fig_technical,
+        fig_prediction,
+        predictions,
+    ) = analyze_stock(symbol, prediction_days)
+
+    if not fundamental_info:
+        return (
+            "Data unavailable.",
+            gr.Plot.update(value=None),
+            gr.Plot.update(value=None),
+            gr.Plot.update(value=None),
+            "",
+            "",
+            "",
+            "",
         )
-        
-        # Load initial analysis
-        demo.load(
-            fn=update_analysis,
-            inputs=[stock_selector, prediction_days],
-            outputs=[
-                company_name, current_price, market_cap, pe_ratio, dividend_yield, volume, fundamentals_text,
-                rsi_value, macd_signal, bb_position, overall_signal, signal_strength, signals_text,
-                support_level, resistance_level, stop_loss, predicted_high, predicted_low, predicted_change,
-                prediction_summary, price_chart, technical_chart, prediction_chart
-            ]
+
+    summary_text = f"""
+**{fundamental_info.get('name', 'N/A')} ({symbol})**
+- Current Price: Rp{fundamental_info.get('current_price', 0):,.2f}
+- Market Cap: {fundamental_info.get('market_cap', 0):,}
+- P/E Ratio: {fundamental_info.get('pe_ratio', 0):.2f}
+- Dividend Yield: {fundamental_info.get('dividend_yield', 0):.2f}%
+- Volume: {fundamental_info.get('volume', 0):,}
+"""
+
+    signal_text = f"""
+### Overall Signal: **{signals.get('overall', 'N/A')}**
+Strength: {signals.get('strength', 0):.2f}%
+Support: {signals.get('support', 0):,.2f}
+Resistance: {signals.get('resistance', 0):,.2f}
+Stop Loss: {signals.get('stop_loss', 0):,.2f}
+
+**Signal Details:**
+{signals.get('details', '')}
+"""
+
+    prediction_text = f"""
+### AI Forecast (Chronos-Bolt)
+Predicted High (30d): {predictions.get('high_30d', 0):,.2f}
+Predicted Low (30d): {predictions.get('low_30d', 0):,.2f}
+Expected Change (%): {predictions.get('change_pct', 0):.2f}%
+Prediction Analysis:
+{predictions.get('summary', 'No analysis available')}
+"""
+
+    return (
+        summary_text,
+        fig_price,
+        fig_technical,
+        fig_prediction,
+        signal_text,
+        f"{predictions.get('high_30d', 0):,.2f}",
+        f"{predictions.get('low_30d', 0):,.2f}",
+        f"{predictions.get('change_pct', 0):.2f}%",
+        prediction_text,
+    )
+
+stock_options = get_indonesian_stocks()
+
+with gr.Blocks(title="AI Stock Analysis (Chronos-Bolt Edition)", theme="soft") as app:
+    gr.Markdown("# 📊 AI Stock Analysis Dashboard (Chronos-Bolt Base Model)")
+
+    with gr.Row():
+        symbol = gr.Dropdown(
+            label="Select Stock",
+            choices=list(stock_options.keys()),
+            value="BBCA.JK",
+            interactive=True,
+        )
+        prediction_days = gr.Slider(
+            label="Prediction Period (Days)",
+            minimum=5,
+            maximum=60,
+            step=5,
+            value=30,
+            interactive=True,
         )
-    
-    return demo
+
+    analyze_button = gr.Button("🔍 Analyze Stock")
+
+    with gr.Tab("Fundamentals & Signals"):
+        fundamentals_output = gr.Markdown()
+        signal_output = gr.Markdown()
+
+    with gr.Tab("Charts"):
+        price_chart = gr.Plot()
+        technical_chart = gr.Plot()
+        prediction_chart = gr.Plot()
+
+    with gr.Tab("Predictions"):
+        predicted_high = gr.Textbox(label="Predicted High (30d)")
+        predicted_low = gr.Textbox(label="Predicted Low (30d)")
+        predicted_change = gr.Textbox(label="Expected Change (%)")
+        prediction_summary = gr.Markdown(label="Prediction Analysis")
+
+    analyze_button.click(
+        fn=update_analysis,
+        inputs=[symbol, prediction_days],
+        outputs=[
+            fundamentals_output,
+            price_chart,
+            technical_chart,
+            prediction_chart,
+            signal_output,
+            predicted_high,
+            predicted_low,
+            predicted_change,
+            prediction_summary,
+        ],
+    )
 
 if __name__ == "__main__":
-    demo = create_ui()
-    demo.launch()
+    app.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=True)