Upload folder using huggingface_hub

app.py

@@ -248,13 +248,17 @@ def get_ai_response(question, stock_data, api_key):
     try:
         client = anthropic.Anthropic(api_key=api_key)
         
-        # Prepare context
+        # Prepare context with current stock data
+        latest_price = stock_data['Close'].iloc[-1]
+        prev_price = stock_data['Close'].iloc[-2] if len(stock_data) > 1 else latest_price
+        daily_change = ((latest_price / prev_price) - 1) * 100 if prev_price != 0 else 0
+        
         context = f"""
         You are a financial analyst AI assistant specializing in CoreWeave (CRWV) stock analysis.
         
         Current stock data available:
-        - Latest Close Price: ${stock_data['Close'].iloc[-1]:.2f}
-        - Daily Change: {((stock_data['Close'].iloc[-1] / stock_data['Close'].iloc[-2]) - 1) * 100:.2f}%
+        - Latest Close Price: ${latest_price:.2f}
+        - Daily Change: {daily_change:.2f}%
         - Volume: {stock_data['Volume'].iloc[-1]:,}
         - 52-week High: ${stock_data['High'].max():.2f}
         - 52-week Low: ${stock_data['Low'].min():.2f}
@@ -266,18 +270,43 @@ def get_ai_response(question, stock_data, api_key):
         clearly state your limitations.
         """
         
-        message = client.messages.create(
-            model="claude-3-sonnet-20240229",
-            max_tokens=1000,
-            temperature=0.7,
-            system=context,
-            messages=[{"role": "user", "content": question}]
-        )
+        # Try multiple model names in order of preference
+        models_to_try = [
+            "claude-3-5-sonnet-20241022",  # Latest Sonnet 3.5
+            "claude-3-5-sonnet-20240620",  # Previous Sonnet 3.5
+            "claude-3-sonnet-20240229",    # Original Sonnet 3
+            "claude-3-haiku-20240307"      # Fallback to Haiku
+        ]
         
-        return message.content[0].text
+        for model_name in models_to_try:
+            try:
+                message = client.messages.create(
+                    model=model_name,
+                    max_tokens=1000,
+                    temperature=0.7,
+                    system=context,
+                    messages=[{"role": "user", "content": question}]
+                )
+                return message.content[0].text
+            
+            except Exception as model_error:
+                if "not_found_error" in str(model_error):
+                    continue  # Try next model
+                else:
+                    return f"Error with model {model_name}: {str(model_error)}"
+        
+        return "Unable to connect to AI service. Please check your API key or try again later."
         
     except Exception as e:
-        return f"Error getting AI response: {str(e)}"
+        error_msg = str(e)
+        if "authentication" in error_msg.lower():
+            return "❌ Invalid API key. Please check your Anthropic API key and try again."
+        elif "rate_limit" in error_msg.lower():
+            return "⏳ Rate limit exceeded. Please wait a moment and try again."
+        elif "insufficient" in error_msg.lower():
+            return "💳 Insufficient credits. Please check your Anthropic account balance."
+        else:
+            return f"❌ AI service error: {error_msg}"
 
 # Main app
 def main():
@@ -411,6 +440,12 @@ def main():
     st.markdown("---")
     st.subheader("🤖 AI Stock Analyst")
     
+    # API Key status
+    if api_key:
+        st.success("✅ API key provided - AI features enabled")
+    else:
+        st.warning("⚠️ Please enter your Anthropic API key in the sidebar to enable AI chat")
+    
     # Chat interface
     chat_container = st.container()
     
@@ -431,13 +466,14 @@ def main():
     # Chat input
     user_question = st.text_input(
         "Ask me anything about CoreWeave stock:",
-        placeholder="e.g., What's your analysis of the current price trend?"
+        placeholder="e.g., What's your analysis of the current price trend?",
+        disabled=not api_key
     )
     
     col_send, col_clear = st.columns([1, 1])
     
     with col_send:
-        if st.button("Send", type="primary") and user_question:
+        if st.button("Send", type="primary", disabled=not api_key) and user_question:
             # Add user message to history
             st.session_state.chat_history.append({
                 'role': 'user',
@@ -474,7 +510,7 @@ def main():
     cols = st.columns(len(sample_questions))
     for i, question in enumerate(sample_questions):
         with cols[i]:
-            if st.button(question, key=f"sample_{i}"):
+            if st.button(question, key=f"sample_{i}", disabled=not api_key):
                 st.session_state.chat_history.append({'role': 'user', 'content': question})
                 ai_response = get_ai_response(question, hist_data, api_key)
                 st.session_state.chat_history.append({'role': 'assistant', 'content': ai_response})