Update app.py

app.py

@@ -53,7 +53,14 @@ if submit_button:
     st.pyplot(fig)
 
     df.reset_index(inplace=True)
-    df.drop(['Date', 'Adj Close'], axis=1, inplace=True)
+
+    # Determine which columns to drop based on their existence in the DataFrame
+    cols_to_drop = ['Date']
+    if 'Adj Close' in df.columns:
+        cols_to_drop.append('Adj Close')
+    
+    df.drop(cols_to_drop, axis=1, inplace=True)
+
     
 
     # Let's understand the 100 days moving average along with 200 days moving average