update app

app.py

@@ -36,7 +36,6 @@ def load_model():
         device_map="auto",
         trust_remote_code=True
     )
-    # CRITICAL FIX: Skip tokenizer loading
     tokenizer = None 
     return model, tokenizer
 
@@ -57,6 +56,10 @@ def get_stock_data(symbol, period="1y"):
 
 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"
+        
     # Get stock data
     data, stock = get_stock_data(symbol)
     
@@ -76,11 +79,11 @@ def analyze_stock(symbol, prediction_days=30):
     predictions = predict_prices(data, model, tokenizer, prediction_days)
     
     # Create charts
-    fig_price = create_price_chart(data, indicators) # RENAMED
-    fig_technical = create_technical_chart(data, indicators) # RENAMED
-    fig_prediction = create_prediction_chart(data, predictions) # RENAMED
+    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 # Returning renamed local variables
+    return fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction
 
 def create_ui():
     """Create the Gradio interface"""
@@ -120,11 +123,11 @@ def create_ui():
         
         with gr.Row():
             with gr.Column(scale=2):
-                stock_selector = gr.Dropdown(
-                    choices=list(IDX_STOCKS.keys()),
-                    value="BBCA.JK",
-                    label="📊 Select Indonesian Stock",
-                    info="Choose from top IDX stocks"
+                # 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 (Tidak perlu menambahkan .JK)"
                 )
                 
                 with gr.Row():
@@ -211,38 +214,37 @@ def create_ui():
         
         # Event handlers
         def update_analysis(symbol, pred_days):
-            fundamental_info, indicators, signals, fig_price, fig_technical, fig_prediction = analyze_stock(symbol, pred_days) # Receives renamed local variables
+            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: "Error loading data",
+                    company_name: "Gagal memuat data",
                     current_price: 0,
                     market_cap: "N/A",
                     pe_ratio: 0,
                     dividend_yield: 0,
                     volume: 0,
-                    fundamentals_text: "Unable to fetch stock data. Please try another symbol.",
+                    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: "No signals available",
+                    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: "No predictions available",
-                    price_chart: None,
-                    technical_chart: None,
-                    prediction_chart: None
+                    prediction_summary: "Prediksi tidak tersedia",
+                    price_chart: go.Figure(),
+                    technical_chart: go.Figure(),
+                    prediction_chart: go.Figure()
                 }
             
             # 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),