Update app.py

app.py

@@ -100,6 +100,43 @@ def detect_chart_type(client, image_data):
         st.error(f"Error in chart type detection: {str(e)}")
         return "Other"
 
+def analyze_chart_with_claude(client, image_data, prompt, chart_type=None):
+    """Analyze chart using Claude Vision"""
+    try:
+        encoded_image = base64.b64encode(image_data).decode('utf-8')
+        
+        # If chart type wasn't provided, detect it first
+        if not chart_type:
+            chart_type = detect_chart_type(client, image_data)
+            st.info(f"Detected chart type: {chart_type}")
+        
+        message = client.messages.create(
+            model="claude-3-opus-20240229",
+            max_tokens=1000,
+            messages=[{
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": prompt.format(chart_type=chart_type)
+                    },
+                    {
+                        "type": "image",
+                        "source": {
+                            "type": "base64",
+                            "media_type": "image/jpeg",
+                            "data": encoded_image
+                        }
+                    }
+                ]
+            }]
+        )
+        
+        return message.content[0].text, chart_type
+    except Exception as e:
+        st.error(f"Error in Claude analysis: {str(e)}")
+        return None, None
+
 def continue_analysis_with_claude(client, question, previous_analysis, image_data=None):
     """Continue the analysis based on a follow-up question"""
     try:
@@ -282,16 +319,17 @@ def main():
             st.title("🚀 Chart Analysis AI")
             upload_option = st.radio(
                 "Choose input method:",
-                ("Upload Image", "Take Screenshot", "Ask Question")
+                ("Upload Image", "Take Screenshot", "Ask Question"),
+                key="analysis_upload_option"  # Added unique key
             )
             
             # File uploader
             if upload_option == "Upload Image":
-                uploaded_file = st.file_uploader("Upload your chart", type=["png", "jpg", "jpeg"])
+                uploaded_file = st.file_uploader("Upload your chart", type=["png", "jpg", "jpeg"], key="analysis_file_uploader")
                 if uploaded_file:
                     st.session_state.current_image = uploaded_file.getvalue()
             elif upload_option == "Take Screenshot":
-                if st.button("Take Screenshot", key="screenshot"):
+                if st.button("Take Screenshot", key="analysis_screenshot_button"):
                     st.info("Feature coming soon! For now, please use the Upload Image option.")
                     screenshot_taken = False
             
@@ -300,13 +338,15 @@ def main():
             patterns = st.multiselect(
                 "Patterns to Look For",
                 ["Double Top/Bottom", "Head and Shoulders", "Triangle", 
-                 "Flag", "Wedge", "Channel", "Support/Resistance"]
+                 "Flag", "Wedge", "Channel", "Support/Resistance"],
+                key="analysis_patterns"
             )
             
             indicators = st.multiselect(
                 "Technical Indicators",
                 ["Moving Averages", "RSI", "MACD", "Bollinger Bands", 
-                 "Volume", "Stochastic", "ADX"]
+                 "Volume", "Stochastic", "ADX"],
+                key="analysis_indicators"
             )
 
         # Main content area
@@ -396,8 +436,8 @@ def main():
                         st.write(f"Follow-up: {chat['question']}")
             
             # Save chat options
-            save_name = st.text_input("Save chat as (optional):")
-            if st.button("Save Chat"):
+            save_name = st.text_input("Save chat as (optional):", key="save_chat_name")
+            if st.button("Save Chat", key="save_chat_button"):
                 if st.session_state.chat_history:
                     filename = save_chat_history(
                         st.session_state.chat_history,
@@ -412,8 +452,8 @@ def main():
         st.title("📚 Learn Trading")
         
         # Search or select trading concept
-        concept = st.text_input("Enter a trading concept you'd like to learn about (e.g., 'evening star pattern', 'RSI', 'MACD'):")
-        if st.button("Learn"):
+        concept = st.text_input("Enter a trading concept you'd like to learn about (e.g., 'evening star pattern', 'RSI', 'MACD'):", key="learn_concept")
+        if st.button("Learn", key="learn_button"):
             if concept:
                 with st.spinner("Getting educational content..."):
                     education_content = get_trading_education(client, concept)