Update app.py

app.py

@@ -1,62 +1,69 @@
 import gradio as gr
-from chronos import ChronosBoltPipeline
+from chronos import ChronosBoltPipeline  # Corrected Import
 import torch
 import numpy as np
 
-# Load the Brain
-pipeline = Chronos2Pipeline.from_pretrained("amazon/chronos-bolt-tiny")
+# 1. LOAD THE BRAIN (Matched to the correct model engine)
+pipeline = ChronosBoltPipeline.from_pretrained("amazon/chronos-bolt-tiny")
 
 def analyze_trade(prices_string, rr_ratio):
     try:
-        # 1. Process Input
+        # 1. Clean and Process Input
         prices = [float(x.strip()) for x in prices_string.split(",")]
         current_price = prices[-1]
-        context = torch.tensor(prices)
+        
+        # Chronos-Bolt expects [Batch, Sequence] shape
+        context = torch.tensor(prices).unsqueeze(0)
         
         # 2. Run AI Prediction
-        # We fetch 0.1, 0.5 (median), and 0.9 quantiles
+        # length 20 = roughly 2 business weeks
         forecast = pipeline.predict(context, prediction_length=20)
         
-        p10 = forecast.quantile(0.1).item()
-        p50 = forecast.quantile(0.5).item() # The "Most Likely" path
-        p90 = forecast.quantile(0.9).item()
+        # 3. Calculate Quantiles (Bolt-Tiny outputs raw samples)
+        # forecast[0] is the matrix of possible futures
+        p10 = np.quantile(forecast[0].numpy(), 0.1)
+        p50 = np.quantile(forecast[0].numpy(), 0.5)
+        p90 = np.quantile(forecast[0].numpy(), 0.9)
         
-        # 3. CALCULATE CERTAINTY (Confidence)
-        # If p90 and p10 are close together, the AI is very certain.
-        # If they are far apart, the AI is confused (high volatility).
+        # 4. CERTAINTY (Confidence)
+        # Narrow gap = AI is certain. Wide gap = High Market Noise.
         volatility = (p90 - p10) / p50
         certainty = max(0, min(100, 100 * (1 - volatility)))
         
-        # 4. RR STRATEGY LOGIC
-        # Risk is the distance to the AI "Floor" (p10)
+        # 5. DYNAMIC TRADING LOGIC
+        # Floor (Stop Loss) is the P10
         risk_amount = current_price - p10
+        
         if risk_amount <= 0:
-            return "Inconclusive (Price below Floor)", 0, 0, 0, "0%"
+            return "Wait for Dip (Price < Floor)", "N/A", "0%", "0%", "Low Probability"
 
-        # Take Profit is Risk * RR Ratio
+        # TP = Current Price + (Risk * RR Ratio)
         target_tp = current_price + (risk_amount * rr_ratio)
         
-        # 5. POTENTIAL OUTCOME
-        potential_gain_pct = ((target_tp - current_price) / current_price) * 100
-        potential_loss_pct = ((current_price - p10) / current_price) * 100
+        # Potential Gain/Loss Percentages
+        gain_pct = ((target_tp - current_price) / current_price) * 100
+        loss_pct = ((current_price - p10) / current_price) * 100
+
+        # Confidence Label
+        status = " HIGH" if certainty > 75 else " MED" if certainty > 50 else " LOW"
 
         return (
-            f"${p10:.2f}",          # Stop Loss
-            f"${target_tp:.2f}",    # Take Profit
-            f"{potential_gain_pct:.2f}%", # Potential Win
-            f"{potential_loss_pct:.2f}%", # Potential Loss
-            f"{certainty:.1f}%"     # AI CERTAINTY
+            f"${p10:.2f}", 
+            f"${target_tp:.2f}", 
+            f"+{gain_pct:.2f}%", 
+            f"-{loss_pct:.2f}%", 
+            f"{certainty:.1f}% ({status})"
         )
 
     except Exception as e:
         return f"Error: {str(e)}", "", "", "", ""
 
-# 6. GRADIO INTERFACE (The "Hidden" API + Simple Dashboard)
+# 6. INTERFACE SETUP
 demo = gr.Interface(
     fn=analyze_trade,
     inputs=[
-        gr.Textbox(label="Stock Price History (Last 250 days)", placeholder="150.1, 150.5..."),
-        gr.Slider(minimum=1.0, maximum=5.0, value=2.0, label="Risk/Reward Ratio (Desired)")
+        gr.Textbox(label="Stock History (Comma Separated)", placeholder="150, 152, 151, 155..."),
+        gr.Slider(minimum=1.0, maximum=5.0, value=2.0, label="Risk/Reward Ratio")
     ],
     outputs=[
         gr.Text(label="Suggested Stop Loss"),
@@ -65,7 +72,7 @@ demo = gr.Interface(
         gr.Text(label="Potential Loss if Stopped"),
         gr.Text(label="AI Confidence Level")
     ],
-    title="Chronos Trading Intelligence"
+    title="Chronos-Bolt Live Analytics"
 )
 
 demo.launch()