Create ui/components.py

ui/components.py

@@ -0,0 +1,80 @@
+# src/ui/components.py
+
+import streamlit as st
+from ..config.settings import CHART_PATTERNS, TECHNICAL_INDICATORS
+
+def create_sidebar():
+    """Create the sidebar with analysis options"""
+    with st.sidebar:
+        st.title("🚀 Chart Analysis AI")
+        upload_option = st.radio(
+            "Choose input method:",
+            ("Upload Image", "Take Screenshot", "Ask Question"),
+            key="sidebar_upload_method"
+        )
+        
+        uploaded_file = None
+        if upload_option == "Upload Image":
+            uploaded_file = st.file_uploader(
+                "Upload your chart",
+                type=["png", "jpg", "jpeg"],
+                key="sidebar_file_uploader"
+            )
+        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.subheader("Analysis Options")
+        patterns = st.multiselect(
+            "Patterns to Look For",
+            CHART_PATTERNS,
+            key="sidebar_patterns"
+        )
+        
+        indicators = st.multiselect(
+            "Technical Indicators",
+            TECHNICAL_INDICATORS,
+            key="sidebar_indicators"
+        )
+        
+        return upload_option, uploaded_file, patterns, indicators
+
+def show_analysis_section(uploaded_file):
+    """Show the main analysis section"""
+    if uploaded_file is not None:
+        st.image(uploaded_file, caption="Uploaded Chart", use_container_width=True)
+    
+    analyze_btn = st.button("Analyze Chart", key="main_analyze_btn")
+    return analyze_btn
+
+def show_chat_history(chat_history):
+    """Display chat history"""
+    st.subheader("Chat History")
+    for idx, chat in enumerate(chat_history):
+        with st.expander(
+            f"Analysis {idx + 1} - {chat.get('timestamp', 'No date')}",
+            key=f"chat_expander_{idx}"
+        ):
+            st.write(chat.get('analysis', ''))
+            if chat.get('question'):
+                st.write(f"Follow-up: {chat['question']}")
+
+def show_follow_up_section(current_analysis):
+    """Show the follow-up question section"""
+    st.subheader("Continue Analysis")
+    follow_up = st.text_input(
+        "Ask a follow-up question:",
+        key="followup_input"
+    )
+    send_btn = st.button("Send", key="followup_send_btn")
+    return follow_up if send_btn else None
+
+def show_save_options():
+    """Show save chat options"""
+    st.subheader("Save Analysis")
+    save_name = st.text_input(
+        "Save chat as (optional):",
+        key="save_chat_name"
+    )
+    save_btn = st.button("Save", key="save_chat_btn")
+    return save_name if save_btn else None