Update app.py

app.py

@@ -54,7 +54,6 @@ def calculate_basic_indicators(data: pd.DataFrame) -> pd.DataFrame:
     
     return df.ffill().bfill()
 
-# Also update the system prompt in generate_analysis_prompt to ensure structured output:
 @pxt.udf
 def generate_analysis_prompt(data: str, analysis_type: str) -> list[dict]:
     """Generate a structured prompt for AI analysis"""
@@ -112,7 +111,6 @@ def parse_analysis_response(response: str) -> Dict[str, str]:
     for line in response.split('\n'):
         line = line.strip()
         
-        # Check if this line is a section header (now handling markdown formatting)
         matched_section = None
         for section in sections.keys():
             # Remove asterisks and check for exact match
@@ -133,11 +131,9 @@ def parse_analysis_response(response: str) -> Dict[str, str]:
             if cleaned_content:  # Only add non-empty lines
                 buffer.append(cleaned_content)
     
-    # Save the last section
     if current_section and buffer:
         sections[current_section] = '\n'.join(buffer).strip()
     
-    # Clean up sections and provide meaningful defaults
     section_messages = {
         'SUMMARY': 'Market analysis summary not available',
         'TECHNICAL ANALYSIS': 'Technical analysis not available',
@@ -147,7 +143,6 @@ def parse_analysis_response(response: str) -> Dict[str, str]:
         'RECOMMENDATION': 'Investment recommendation not available'
     }
     
-    # Only use default messages if section is truly empty
     for key in sections:
         if sections[key] is None or not sections[key].strip():
             sections[key] = section_messages[key]
@@ -163,7 +158,6 @@ def create_visualization(data: pd.DataFrame, technical_depth: str) -> go.Figure:
         subplot_titles=('Price & Moving Averages', 'Volume', 'RSI' if technical_depth == 'advanced' else None)
     )
 
-    # Price candlesticks with improved styling
     fig.add_trace(
         go.Candlestick(
             x=data.index,
@@ -178,7 +172,6 @@ def create_visualization(data: pd.DataFrame, technical_depth: str) -> go.Figure:
         row=1, col=1
     )
 
-    # Moving averages with distinct colors
     colors = {'MA20': '#1E88E5', 'MA50': '#FFC107', 'MA200': '#7B1FA2'}
     for ma, color in colors.items():
         fig.add_trace(
@@ -191,7 +184,6 @@ def create_visualization(data: pd.DataFrame, technical_depth: str) -> go.Figure:
             row=1, col=1
         )
 
-    # Volume with color based on price change
     colors = ['#26A69A' if close >= open_price else '#EF5350' 
          for close, open_price in zip(data['Close'].values, data['Open'].values)]
     fig.add_trace(
@@ -215,7 +207,6 @@ def create_visualization(data: pd.DataFrame, technical_depth: str) -> go.Figure:
             row=3, col=1
         )
         
-        # Add RSI reference lines
         fig.add_hline(y=70, line_dash="dash", line_color="red", row=3, col=1)
         fig.add_hline(y=30, line_dash="dash", line_color="green", row=3, col=1)
 
@@ -232,7 +223,6 @@ def create_visualization(data: pd.DataFrame, technical_depth: str) -> go.Figure:
         )
     )
     
-    # Update y-axes labels
     fig.update_yaxes(title_text="Price", row=1, col=1)
     fig.update_yaxes(title_text="Volume", row=2, col=1)
     if technical_depth == 'advanced':
@@ -281,7 +271,6 @@ def process_outputs(ticker_symbol, analysis_type, time_horizon, risk_tolerance,
             max_tokens=1000
         )
 
-        # Process the analysis with better error handling
         try:
             analysis_text = data_table.select(
                 analysis=data_table.analysis.choices[0].message.content
@@ -310,7 +299,6 @@ def process_outputs(ticker_symbol, analysis_type, time_horizon, risk_tolerance,
             parsed_analysis = parse_analysis_response("")
             raw_llm_output = f"Error processing analysis: {str(analysis_error)}"
 
-        # Prepare market stats with proper number formatting
         try:
             current_price = float(technical_data['Close'].iloc[-1])
             previous_price = float(technical_data['Close'].iloc[-2])
@@ -327,7 +315,6 @@ def process_outputs(ticker_symbol, analysis_type, time_horizon, risk_tolerance,
             'RSI': f"{rsi:.2f}"
         }
 
-        # Add timestamp to technical data
         technical_data_with_time = technical_data.reset_index()
         technical_data_with_time['Date'] = technical_data_with_time['Date'].dt.strftime('%Y-%m-%d %H:%M:%S')
 
@@ -379,7 +366,6 @@ def create_interface() -> gr.Blocks:
             """
         )
 
-        # Information Accordions
         with gr.Row():
             with gr.Column():
                 with gr.Accordion("🎯 What does it do?", open=False):
@@ -428,7 +414,7 @@ def create_interface() -> gr.Blocks:
         )
 
         with gr.Row():
-            # Left sidebar for inputs (reduced width)
+            # Left sidebar for inputs
             with gr.Column(scale=1):
                 with gr.Row():
                     gr.Markdown("### 📊 Analysis Parameters")
@@ -516,13 +502,12 @@ def create_interface() -> gr.Blocks:
                 with gr.Row():
                     plot_output = gr.Plot()
 
-        # AI Analysis section with better layout
+        # AI Analysis section
         with gr.Row():
             with gr.Column(scale=2):
                 with gr.Row():
                     gr.Markdown("### 🤖 AI Analysis")
                     
-                # Summary at the top
                 with gr.Row():
                     summary = gr.Textbox(
                         label="Executive Summary",
@@ -531,7 +516,6 @@ def create_interface() -> gr.Blocks:
                         show_label=True
                     )
                     
-                # Main analysis sections
                 with gr.Row():
                     with gr.Column(scale=1):
                         tech_analysis = gr.Textbox(
@@ -561,7 +545,6 @@ def create_interface() -> gr.Blocks:
                             show_label=True
                         )
                 
-                # Recommendation at the bottom
                 with gr.Row():
                     recommendation = gr.Textbox(
                         label="Investment Recommendation",
@@ -570,7 +553,6 @@ def create_interface() -> gr.Blocks:
                         show_label=True
                     )
         
-        # Examples section at the bottom
         gr.Examples(
             examples=[
                 ["AAPL", "comprehensive", "medium", "moderate", "balanced", "advanced"],