Update ui/components.py

ui/components.py

@@ -9,20 +9,23 @@ def create_sidebar():
         st.title("🚀 Chart Analysis AI")
         upload_option = st.radio(
             "Choose input method:",
-            ("Upload Image", "Take Screenshot", "Ask Question"),
+            ("Upload Images", "Take Screenshot", "Ask Question"),
             key="sidebar_upload_method"
         )
         
-        uploaded_file = None
-        if upload_option == "Upload Image":
-            uploaded_file = st.file_uploader(
-                "Upload your chart",
+        uploaded_files = None
+        if upload_option == "Upload Images":
+            uploaded_files = st.file_uploader(
+                "Upload your charts",
                 type=["png", "jpg", "jpeg"],
+                accept_multiple_files=True,  # Allow multiple files
                 key="sidebar_file_uploader"
             )
+            if uploaded_files:
+                st.info(f"Uploaded {len(uploaded_files)} charts")
         elif upload_option == "Take Screenshot":
             if st.button("Take Screenshot", key="sidebar_screenshot_btn"):
-                st.info("Feature coming soon! For now, please use the Upload Image option.")
+                st.info("Feature coming soon! For now, please use the Upload Images option.")
         
         st.subheader("Analysis Options")
         patterns = st.multiselect(
@@ -36,15 +39,35 @@ def create_sidebar():
             TECHNICAL_INDICATORS,
             key="sidebar_indicators"
         )
+
+        # Add comparison options
+        if uploaded_files and len(uploaded_files) > 1:
+            st.subheader("Comparison Options")
+            comparison_type = st.selectbox(
+                "Comparison Type",
+                ["Individual Analysis", "Correlated Analysis", "Market Trend Analysis"],
+                key="comparison_type"
+            )
+            st.info("""
+            - Individual Analysis: Analyze each chart separately
+            - Correlated Analysis: Find relationships between charts
+            - Market Trend Analysis: Identify broader market patterns
+            """)
+        else:
+            comparison_type = "Individual Analysis"
         
-        return upload_option, uploaded_file, patterns, indicators
+        return upload_option, uploaded_files, patterns, indicators, comparison_type
 
-def show_analysis_section(uploaded_file):
+def show_analysis_section(uploaded_files):
     """Show the main analysis section"""
-    if uploaded_file is not None:
-        st.image(uploaded_file, caption="Uploaded Chart", use_container_width=True)
+    if uploaded_files:
+        # Create a grid layout for multiple images
+        cols = st.columns(min(len(uploaded_files), 2))  # Max 2 columns
+        for idx, uploaded_file in enumerate(uploaded_files):
+            with cols[idx % 2]:
+                st.image(uploaded_file, caption=f"Chart {idx + 1}", use_container_width=True)
     
-    analyze_btn = st.button("Analyze Chart", key="main_analyze_btn")
+    analyze_btn = st.button("Analyze Charts", key="main_analyze_btn")
     return analyze_btn
 
 def show_chat_history(chat_history):