Update app.py

app.py

@@ -88,7 +88,12 @@ def filter_stocks_by_indicators(stocks, start_date, end_date, rsi_threshold, sto
                    (indicator_choice == 'StochRSI' and stoch_rsi_value < stoch_rsi_threshold) or \
                    (indicator_choice == 'OBV' and obv_change > obv_threshold) or \
                    (indicator_choice == 'All' and rsi_value < rsi_threshold and stoch_rsi_value < stoch_rsi_threshold and obv_change > obv_threshold):
-                    filtered_stocks.append(ticker)
+                    filtered_stocks.append({
+                        'Ticker': ticker,
+                        'RSI': rsi_value,
+                        'StochRSI': stoch_rsi_value,
+                        'OBV_Change': obv_change
+                    })
                     st.info(f"{ticker} - RSI: {rsi_value:.2f}, StochRSI: {stoch_rsi_value:.2f}, OBV Change: {obv_change:.2f}%")
                 else:
                     st.info(f"{ticker} - RSI: {rsi_value:.2f}, StochRSI: {stoch_rsi_value:.2f}, OBV Change: {obv_change:.2f}% (doesn't meet criteria)")
@@ -133,15 +138,17 @@ def main():
         
         if filtered_stocks:
             st.success(f"Found {len(filtered_stocks)} stocks meeting the criteria:")
-            st.write(filtered_stocks)
-            df = pd.DataFrame(filtered_stocks, columns=["Ticker"])
+            df = pd.DataFrame(filtered_stocks)
+            st.dataframe(df)
+            
+            csv_data = df.to_csv(index=False).encode('utf-8')
             st.download_button(
                 label="Download CSV",
-                data=df.to_csv(index=False).encode('utf-8'),
+                data=csv_data,
                 file_name=output_filename,
                 mime='text/csv',
             )
-            st.success(f"Click the button above to download the filtered stocks.")
+            st.success(f"Click the button above to download the filtered stocks with RSI, StochRSI, and OBV Change data.")
         else:
             st.warning(f"No stocks found meeting the selected criteria.")