Update app.py

app.py

@@ -5,6 +5,12 @@ import requests
 import json
 import os
 
+def verify_api_key():
+    api_key = os.environ.get('MISTRAL_API_KEY')
+    if not api_key:
+        return "ERROR: Mistral API key not found in environment variables"
+    return "API key found"
+
 def fetch_nifty_data():
     try:
         end_date = datetime.now()
@@ -41,6 +47,10 @@ def fetch_nifty_data():
 
 def get_mistral_analysis(df):
     try:
+        api_key = os.environ.get('MISTRAL_API_KEY')
+        if not api_key:
+            return "Error: Mistral API key not found. Please add it to the environment variables."
+
         # Get last 10 days of data
         last_10_days = df.tail(10)
         
@@ -56,25 +66,36 @@ def get_mistral_analysis(df):
         
         headers = {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {os.environ.get('MISTRAL_API_KEY')}"
+            "Authorization": f"Bearer {api_key}"
         }
         
         data = {
-            "model": "mistral-tiny",
-            "messages": [{"role": "user", "content": data_str}]
+            "model": "mistral-small",  # Changed from mistral-tiny to mistral-small
+            "messages": [{"role": "user", "content": data_str}],
+            "temperature": 0.7,
+            "max_tokens": 500
         }
         
+        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:
-            return f"Error: {response.status_code} - {response.text}"
+            return f"Error {response.status_code}: {response.text}"
             
     except Exception as e:
         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
+    
     df = fetch_nifty_data()
     analysis = get_mistral_analysis(df) if not df.empty else "Unable to fetch data"
     return df, analysis
@@ -84,6 +105,9 @@ 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")