Create app.py

app.py

@@ -0,0 +1,244 @@
+import streamlit as st
+import base64
+from datetime import datetime
+import json
+import os
+from openai import OpenAI
+import requests
+from PIL import Image
+import io
+from huggingface_hub import HfApi
+import yaml
+
+def load_huggingface_secret():
+    """Load OpenAI API key from Hugging Face secrets"""
+    try:
+        hf_api = HfApi()
+        # Fetch the secret from Hugging Face
+        secret = hf_api.get_secret('openai_api_key')
+        return secret
+    except Exception as e:
+        st.error(f"Error loading secret from Hugging Face: {str(e)}")
+        return None
+
+def initialize_openai_client():
+    """Initialize OpenAI client with API key from Hugging Face"""
+    api_key = load_huggingface_secret()
+    if api_key:
+        return OpenAI(api_key=api_key)
+    return None
+
+def create_prompt_template(chart_type, patterns, indicators):
+    """Creates a structured prompt for the LLM based on the chart and analysis needs"""
+    prompt = f"""You are an expert financial analyst. Please analyze this {chart_type} chart and provide insights in the following structured format:
+
+    1. VISUAL ANALYSIS
+    - Identify and describe the main trend
+    - Note key price levels visible in the chart
+    - Describe any significant patterns: {', '.join(patterns) if patterns else 'all visible patterns'}
+    - Comment on volume trends if visible
+    - Analyze these technical indicators: {', '.join(indicators) if indicators else 'visible indicators'}
+
+    2. TECHNICAL INTERPRETATION
+    - Current market structure and trend strength
+    - Key support and resistance levels with price points
+    - Any visible divergences or convergences
+    - Pattern reliability assessment
+
+    3. RISK ANALYSIS
+    - Potential risk levels
+    - Risk/reward scenarios
+    - Warning signs or red flags
+    - Market context considerations
+
+    4. ACTIONABLE INSIGHTS
+    - Potential trading scenarios
+    - Key price targets
+    - Suggested stop-loss levels
+    - Timeframe considerations
+
+    5. SIMPLIFIED EXPLANATION
+    Provide a 2-3 sentence summary in simple terms for novice traders.
+
+    IMPORTANT: Clearly mark this as AI-generated analysis for informational purposes only.
+    """
+    return prompt
+
+def analyze_chart_with_openai(client, image_data, prompt):
+    """Analyze chart using OpenAI's vision model"""
+    try:
+        # Encode image to base64
+        encoded_image = base64.b64encode(image_data).decode('utf-8')
+        
+        # Create message with image and prompt
+        messages = [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": prompt
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": f"data:image/jpeg;base64,{encoded_image}"
+                        }
+                    }
+                ]
+            }
+        ]
+        
+        # Call OpenAI API
+        response = client.chat.completions.create(
+            model="gpt-4-vision-preview",
+            messages=messages,
+            max_tokens=1000
+        )
+        
+        return response.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error in OpenAI analysis: {str(e)}")
+        return None
+
+def save_chat_history(chat_history, filename=None):
+    """Saves chat history to a JSON file"""
+    if not os.path.exists("chat_histories"):
+        os.makedirs("chat_histories")
+        
+    if filename is None:
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"chat_history_{timestamp}.json"
+    
+    filepath = os.path.join("chat_histories", filename)
+    with open(filepath, "w") as f:
+        json.dump(chat_history, f)
+    return filename
+
+def load_chat_history(filename):
+    """Loads chat history from a JSON file"""
+    filepath = os.path.join("chat_histories", filename)
+    with open(filepath, "r") as f:
+        return json.load(f)
+
+def main():
+    st.set_page_config(
+        page_title="Stock Chart Assistant",
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
+    
+    # Initialize OpenAI client
+    client = initialize_openai_client()
+    if not client:
+        st.error("Failed to initialize OpenAI client. Please check your Hugging Face secrets.")
+        return
+    
+    # Initialize session state for chat history
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+    
+    # Sidebar
+    with st.sidebar:
+        st.title("🚀 Chart Analysis AI")
+        upload_option = st.radio(
+            "Choose input method:",
+            ("Upload Image", "Take Screenshot", "Ask Question")
+        )
+        
+        # File uploader
+        if upload_option == "Upload Image":
+            uploaded_file = st.file_uploader("Upload your chart", type=["png", "jpg", "jpeg"])
+        elif upload_option == "Take Screenshot":
+            st.button("Open Screenshot Tool", key="screenshot")
+            # Screenshot functionality would be implemented here
+        
+        # Analysis Options
+        st.subheader("Analysis Options")
+        chart_type = st.selectbox(
+            "Chart Type",
+            ["Candlestick", "Line", "OHLC", "Area", "Other"]
+        )
+        
+        patterns = st.multiselect(
+            "Patterns to Look For",
+            ["Double Top/Bottom", "Head and Shoulders", "Triangle", 
+             "Flag", "Wedge", "Channel", "Support/Resistance"]
+        )
+        
+        indicators = st.multiselect(
+            "Technical Indicators",
+            ["Moving Averages", "RSI", "MACD", "Bollinger Bands", 
+             "Volume", "Stochastic", "ADX"]
+        )
+
+    # Main content area
+    st.title("📈 Stock Chart Analysis Assistant")
+    
+    # Create two columns for layout
+    col1, col2 = st.columns([2, 1])
+    
+    with col1:
+        if upload_option == "Ask Question":
+            user_question = st.text_input("What would you like to know about your chart?")
+        
+        # Display uploaded image
+        if uploaded_file is not None:
+            st.image(uploaded_file, caption="Uploaded Chart", use_column_width=True)
+        
+        if st.button("Analyze"):
+            if uploaded_file is None and upload_option != "Ask Question":
+                st.warning("Please upload an image or take a screenshot first.")
+            else:
+                with st.spinner("Analyzing chart..."):
+                    # Generate prompt
+                    prompt = create_prompt_template(chart_type, patterns, indicators)
+                    
+                    if uploaded_file:
+                        # Process image and get analysis
+                        analysis_result = analyze_chart_with_openai(
+                            client,
+                            uploaded_file.getvalue(),
+                            prompt
+                        )
+                        
+                        if analysis_result:
+                            # Add to chat history
+                            st.session_state.chat_history.append({
+                                'timestamp': datetime.now().isoformat(),
+                                'chart_type': chart_type,
+                                'analysis': analysis_result
+                            })
+                            
+                            # Display analysis
+                            st.subheader("Analysis Results")
+                            st.write(analysis_result)
+                            
+                            # Risk warning
+                            st.warning(
+                                "⚠️ This analysis is AI-generated and for informational purposes only. "
+                                "Do not make trading decisions solely based on this information."
+                            )
+    
+    with col2:
+        st.subheader("Chat History")
+        
+        # Display chat history
+        for chat in st.session_state.chat_history:
+            with st.expander(f"Analysis from {chat['timestamp'][:16]}"):
+                st.write(chat['analysis'])
+        
+        # Save chat options
+        save_name = st.text_input("Save chat as (optional):")
+        if st.button("Save Chat"):
+            if st.session_state.chat_history:
+                filename = save_chat_history(
+                    st.session_state.chat_history,
+                    f"{save_name}.json" if save_name else None
+                )
+                st.success(f"Chat saved as {filename}")
+            else:
+                st.warning("No chat history to save.")
+
+if __name__ == "__main__":
+    main()