v1111

app.py

@@ -219,22 +219,42 @@ PLOT_FILENAME = 'generated_plot.png'
 
 # Helper functions for data processing
 def safe_get_column(df, column):
+    # Handle case where column is a tuple (e.g., from multi-index)
+    if isinstance(column, tuple):
+        column = column[0]  # Take the first element of the tuple
+    
+    # Convert column to string in case it's not
+    column = str(column)
+    
     # Try exact match first
     if column in df.columns:
         return df[column]
+        
     # Try case-insensitive match
-    col_lower = column.lower()
-    for col in df.columns:
-        if col.lower() == col_lower:
-            return df[col]
+    try:
+        col_lower = column.lower()
+        for col in df.columns:
+            if str(col).lower() == col_lower:
+                return df[col]
+    except (AttributeError, TypeError):
+        pass  # Skip case-insensitive matching if not applicable
+    
     # If not found, try common variations
     variations = {
-        'close': ['Close', 'Adj Close', 'close', 'adj close', 'CLOSE'],
-        'adj close': ['Adj Close', 'adj close', 'ADJ CLOSE', 'Close', 'close', 'CLOSE']
+        'close': ['Close', 'Adj Close', 'close', 'adj close', 'CLOSE', 'Adj. Close'],
+        'adj close': ['Adj Close', 'adj close', 'ADJ CLOSE', 'Close', 'close', 'CLOSE', 'Adj. Close']
     }
-    for var in variations.get(column.lower(), []):
-        if var in df.columns:
-            return df[var]
+    
+    for var_list in variations.values():
+        for var in var_list:
+            if var in df.columns:
+                return df[var]
+    
+    # If still not found, try to find any column containing 'close'
+    for col in df.columns:
+        if 'close' in str(col).lower():
+            return df[col]
+            
     # If still not found, raise a helpful error
     raise KeyError(f"Column '{column}' not found in DataFrame. Available columns: {list(df.columns)}")