Update app.py

app.py

@@ -54,13 +54,48 @@ def get_mistral_analysis(df):
         # Get last 10 days of data
         last_10_days = df.tail(10)
         
+        # Calculate current price (last close)
+        current_price = last_10_days.iloc[-1]['Close']
+        
         # Format data for prompt
-        data_str = "this is the last 10 days data for nifty index, Open\tHigh\tLow\tClose\n"
+        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)
+
+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"{row['Open']}\t{row['High']}\t{row['Low']}\t{row['Close']}\n"
-        
-        data_str += "so i need you to give me the setup for today by priceaction levels"
-        
+            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:
+
+Resistance Levels:
+- R1:
+- R2:
+- R3:
+
+Today's Target Price:
+- Target:
+
+Trading Setup:
+- Direction:
+- Entry Price Range:
+- Stop Loss:
+- Risk/Reward Ratio:
+
+Analysis Explanation:
+(Provide brief explanation of the price action and reasoning for levels)"""
+
         # Mistral API endpoint
         url = "https://api.mistral.ai/v1/chat/completions"
         
@@ -70,18 +105,17 @@ def get_mistral_analysis(df):
         }
         
         data = {
-            "model": "mistral-small",  # Changed from mistral-tiny to mistral-small
-            "messages": [{"role": "user", "content": data_str}],
+            "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": "user", "content": data_str}
+            ],
             "temperature": 0.7,
-            "max_tokens": 500
+            "max_tokens": 1000
         }
         
-        print(f"Sending request to Mistral AI with API key: {api_key[:5]}...")  # Print first 5 chars of API key for debugging
         response = requests.post(url, headers=headers, json=data)
         
-        print(f"Response status code: {response.status_code}")
-        print(f"Response content: {response.text[:200]}...")  # Print first 200 chars of response
-        
         if response.status_code == 200:
             return response.json()['choices'][0]['message']['content']
         else:
@@ -100,27 +134,49 @@ 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() as demo:
-    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()}")
-    
-    # Add refresh button
-    refresh_btn = gr.Button("Refresh Data and Analysis")
-    
-    # Add outputs
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("### Historical Data")
-            output_table = gr.Dataframe()
-        
-        with gr.Column():
-            gr.Markdown("### Mistral AI Analysis")
-            analysis_output = gr.Textbox(label="Price Action Analysis", lines=10)
-    
+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()}")
+        
+        # Add refresh button with custom styling
+        refresh_btn = gr.Button(
+            "Refresh Data and Analysis",
+            variant="primary",
+            size="lg"
+        )
+        
+        # Add outputs in a responsive layout
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### Historical Data")
+                output_table = gr.Dataframe(
+                    headers=["Date", "Open", "High", "Low", "Close", "Volume"],
+                    wrap=True
+                )
+            
+            with gr.Column(scale=1):
+                gr.Markdown("### Mistral AI Analysis")
+                analysis_output = gr.Textbox(
+                    label="Price Action Analysis",
+                    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,
@@ -133,4 +189,5 @@ with gr.Blocks() as demo:
         outputs=[output_table, analysis_output],
     )
 
+# Launch the app
 demo.launch()