Update app.py

app.py

@@ -58,45 +58,34 @@ def get_mistral_analysis(df):
         current_price = last_10_days.iloc[-1]['Close']
         
         # Format data for prompt
-        data_str = f"""Based on the following NIFTY 50 price data for the last 10 days, analyze the price action and provide:
-1. Key support levels (2-3 levels)
-2. Key resistance levels (2-3 levels)
-3. Potential target price for today
-4. Trading setup recommendation (bullish/bearish)
+        data_str = f"""Based on the following NIFTY 50 price data for the last 10 days, provide only support levels, resistance levels, target price, and trading setup. Format exactly as shown below:
 
-Current price: {current_price}
-
-Last 10 days data (Open, High, Low, Close):
-"""
-        # Add data in a readable format
-        for _, row in last_10_days.iterrows():
-            data_str += f"Date: {row['Date']} - Open: {row['Open']}, High: {row['High']}, Low: {row['Low']}, Close: {row['Close']}\n"
-        
-        data_str += """
-Please provide your analysis in the following format:
 Support Levels:
-- S1:
-- S2:
-- S3:
+- S1: [price] (brief reason)
+- S2: [price] (brief reason)
+- S3: [price] (brief reason)
 
 Resistance Levels:
-- R1:
-- R2:
-- R3:
+- R1: [price] (brief reason)
+- R2: [price] (brief reason)
+- R3: [price] (brief reason)
 
 Today's Target Price:
-- Target:
+- Target ([direction]): [price] (brief reason)
 
 Trading Setup:
-- Direction:
-- Entry Price Range:
-- Stop Loss:
-- Risk/Reward Ratio:
+- Direction: [Bullish/Bearish]
+- Entry Price Range: [range]
+- Stop Loss: [price]
+- Risk/Reward Ratio: [ratio]
 
-Analysis Explanation:
-(Provide brief explanation of the price action and reasoning for levels)"""
+Current price: {current_price}
+
+Data:
+"""
+        for _, row in last_10_days.iterrows():
+            data_str += f"{row['Date']}: O:{row['Open']} H:{row['High']} L:{row['Low']} C:{row['Close']}\n"
 
-        # Mistral API endpoint
         url = "https://api.mistral.ai/v1/chat/completions"
         
         headers = {
@@ -107,11 +96,11 @@ Analysis Explanation:
         data = {
             "model": "mistral-small",
             "messages": [
-                {"role": "system", "content": "You are a professional price action trader. Analyze the given market data and provide specific support, resistance levels, and target prices based on price action principles. Be precise with numbers and explain your reasoning."},
+                {"role": "system", "content": "You are a price action trader. Provide only the requested levels and setup information in the exact format specified. Do not add any additional analysis or explanation."},
                 {"role": "user", "content": data_str}
             ],
             "temperature": 0.7,
-            "max_tokens": 1000
+            "max_tokens": 500
         }
         
         response = requests.post(url, headers=headers, json=data)
@@ -125,7 +114,6 @@ Analysis Explanation:
         return f"Error getting analysis: {str(e)}"
 
 def show_nifty_data_and_analysis():
-    # First verify API key
     api_status = verify_api_key()
     if api_status.startswith("ERROR"):
         return pd.DataFrame(), api_status
@@ -134,49 +122,31 @@ def show_nifty_data_and_analysis():
     analysis = get_mistral_analysis(df) if not df.empty else "Unable to fetch data"
     return df, analysis
 
-# Create Gradio interface with custom CSS
-custom_css = """
-.container {max-width: 1200px; margin: auto;}
-.analysis {white-space: pre-wrap; font-family: monospace;}
-"""
-
 # Create Gradio interface
-with gr.Blocks(css=custom_css) as demo:
-    with gr.Column(elem_classes="container"):
-        gr.Markdown("# NIFTY 50 Analysis with Mistral AI")
-        gr.Markdown("Displays the last 15 days of trading data and AI analysis for NIFTY 50 index")
-        
-        # Add API key status
-        api_status = gr.Markdown(f"API Status: {verify_api_key()}")
+with gr.Blocks() as demo:
+    with gr.Column():
+        gr.Markdown("# NIFTY 50 Levels and Setup")
         
-        # Add refresh button with custom styling
-        refresh_btn = gr.Button(
-            "Refresh Data and Analysis",
-            variant="primary",
-            size="lg"
-        )
+        # Add refresh button
+        refresh_btn = gr.Button("Refresh Data", variant="primary")
         
-        # Add outputs in a responsive layout
+        # Add outputs
         with gr.Row():
             with gr.Column(scale=1):
-                gr.Markdown("### Historical Data")
+                gr.Markdown("### Market Data")
                 output_table = gr.Dataframe(
                     headers=["Date", "Open", "High", "Low", "Close", "Volume"],
                     wrap=True
                 )
             
             with gr.Column(scale=1):
-                gr.Markdown("### Mistral AI Analysis")
+                gr.Markdown("### Trading Levels")
                 analysis_output = gr.Textbox(
-                    label="Price Action Analysis",
+                    label="Levels and Setup",
                     lines=15,
                     elem_classes="analysis"
                 )
-        
-        # Add footer
-        gr.Markdown("---")
-        gr.Markdown("Updated every minute • Data source: Yahoo Finance • Analysis: Mistral AI")
-        
+    
     # Set up refresh button click event
     refresh_btn.click(
         fn=show_nifty_data_and_analysis,