Update app.py

app.py

@@ -3,14 +3,8 @@ import pandas as pd
 from datetime import datetime, timedelta
 import requests
 import json
-from mistralai.client import MistralClient
-from mistralai.models.chat_completion import ChatMessage
 import os
 
-# Initialize Mistral AI client
-# Make sure to set MISTRAL_API_KEY in your Hugging Face Space secrets
-client = MistralClient(api_key=os.environ.get('MISTRAL_API_KEY'))
-
 def fetch_nifty_data():
     try:
         end_date = datetime.now()
@@ -57,17 +51,26 @@ def get_mistral_analysis(df):
         
         data_str += "so i need you to give me the setup for today by priceaction levels"
         
-        # Get analysis from Mistral AI
-        messages = [
-            ChatMessage(role="user", content=data_str)
-        ]
+        # Mistral API endpoint
+        url = "https://api.mistral.ai/v1/chat/completions"
+        
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {os.environ.get('MISTRAL_API_KEY')}"
+        }
+        
+        data = {
+            "model": "mistral-tiny",
+            "messages": [{"role": "user", "content": data_str}]
+        }
         
-        response = client.chat(
-            model="mistral-tiny",
-            messages=messages
-        )
+        response = requests.post(url, headers=headers, json=data)
         
-        return response.messages[0].content
+        if response.status_code == 200:
+            return response.json()['choices'][0]['message']['content']
+        else:
+            return f"Error: {response.status_code} - {response.text}"
+            
     except Exception as e:
         return f"Error getting analysis: {str(e)}"
 
@@ -76,7 +79,7 @@ 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 auto-refresh
+# 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")