update app

app.py

@@ -3,7 +3,7 @@ import yfinance as yf
 import pandas as pd
 import numpy as np
 import torch
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer # Keep import for dependency check
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from datetime import datetime, timedelta
 import plotly.graph_objects as go
 import plotly.express as px
@@ -29,14 +29,14 @@ 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. Tokenizer loading is skipped to bypass fatal error."""
+    """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
     )
-    # CRITICAL FIX: Skip tokenizer loading (it is not used in predict_prices anyway)
+    # CRITICAL FIX: Skip tokenizer loading
     tokenizer = None 
     return model, tokenizer
 
@@ -73,15 +73,14 @@ def analyze_stock(symbol, prediction_days=30):
     signals = generate_trading_signals(data, indicators)
     
     # Make predictions using Chronos-Bolt
-    # Passing the (now None) tokenizer argument to maintain compatibility with utils.py signature
     predictions = predict_prices(data, model, tokenizer, prediction_days)
     
     # Create charts
-    price_chart = create_price_chart(data, indicators)
-    technical_chart = create_technical_chart(data, indicators)
-    prediction_chart = create_prediction_chart(data, predictions)
+    fig_price = create_price_chart(data, indicators) # RENAMED
+    fig_technical = create_technical_chart(data, indicators) # RENAMED
+    fig_prediction = create_prediction_chart(data, predictions) # RENAMED
     
-    return fundamental_info, indicators, signals, price_chart, technical_chart, prediction_chart
+    return fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction # Returning renamed local variables
 
 def create_ui():
     """Create the Gradio interface"""
@@ -212,7 +211,7 @@ def create_ui():
         
         # Event handlers
         def update_analysis(symbol, pred_days):
-            fundamental_info, indicators, signals, price_chart, technical_chart, prediction_chart = analyze_stock(symbol, pred_days)
+            fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction = analyze_stock(symbol, pred_days) # Receives renamed local variables
             
             if fundamental_info is None:
                 return {
@@ -242,6 +241,8 @@ def create_ui():
                 }
             
             # Format outputs
+            # CRITICAL FIX: Use the Gradio component objects (left) as keys 
+            # and the Plotly Figure objects (right) as values.
             return {
                 company_name: fundamental_info.get('name', 'N/A'),
                 current_price: fundamental_info.get('current_price', 0),
@@ -263,9 +264,9 @@ def create_ui():
                 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: price_chart,
-                technical_chart: technical_chart,
-                prediction_chart: prediction_chart
+                price_chart: fig_price,
+                technical_chart: fig_technical,
+                prediction_chart: fig_prediction
             }
         
         analyze_btn.click(