Update app.py

app.py

@@ -13,7 +13,7 @@ def load_openai_key():
     """Load OpenAI API key from HuggingFace secrets"""
     try:
         # Try to get from HuggingFace secrets via environment variable
-        api_key = os.getenv('OPENAI_API_KEY')
+        api_key = os.getenv('OPENAI_API_KEY_HF')
         
         if not api_key:
             st.error("OpenAI API key not found in HuggingFace secrets. Please ensure the secret is set as OPENAI_API_KEY_HF in your HuggingFace space.")
@@ -68,9 +68,10 @@ def detect_chart_type(client, image_data):
 
 def create_prompt_template(patterns, indicators):
     """Creates a structured prompt for the LLM based on the chart and analysis needs"""
-    prompt = """You are an expert financial analyst. Please analyze this {chart_type} chart and provide insights in the following structured format:
+    prompt = """You are an expert financial analyst. Please analyze this financial chart (chart type will be detected automatically) and provide insights in the following structured format:
 
     1. VISUAL ANALYSIS
+    - First identify the type of chart (candlestick, line, OHLC, area, etc.)
     - Identify and describe the main trend
     - Note key price levels visible in the chart
     - Describe any significant patterns: {patterns}
@@ -99,11 +100,11 @@ def create_prompt_template(patterns, indicators):
     Provide a 2-3 sentence summary in simple terms for novice traders.
 
     IMPORTANT: Clearly mark this as AI-generated analysis for informational purposes only.
-    """.format(
+    """
+    return prompt.format(
         patterns=', '.join(patterns) if patterns else 'all visible patterns',
         indicators=', '.join(indicators) if indicators else 'visible indicators'
     )
-    return prompt
 
 def analyze_chart_with_openai(client, image_data, prompt, chart_type=None):
     """Analyze chart using OpenAI's GPT-4 model"""
@@ -232,7 +233,7 @@ def main():
         
         # Display uploaded image
         if uploaded_file is not None:
-            st.image(uploaded_file, caption="Uploaded Chart", use_column_width=True)
+            st.image(uploaded_file, caption="Uploaded Chart", use_container_width=True)
         
         if st.button("Analyze"):
             if upload_option == "Ask Question" and user_question: