Update app.py

app.py

@@ -608,6 +608,11 @@ def output_results(start_date, end_date, tickers_string):
     return  fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
             expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns
 
+    ticker_string_concervative = "KO,PG,JNJ,VZ,PFE,MCD,WMT,MMM,SO,DUK"
+    ticker_string_growth = "AAPL,AMZN,TSLA,NVDA,GOOGL,MSFT,NFLX,V,MA,CRM"
+    ticker_string_balanced = "AAPL,JPM,JNJ,PG,KO,MSFT,PFE,VZ,MMM,WMT"
+    ticker_string_agressive = "TSLA,AMZN,NVDA,GOOGL,SHOP,SQ,MRNA,ZM,SNOW,PTON"
+
 
 # Interface =============================================
 
@@ -626,22 +631,27 @@ with gr.Blocks() as demo:
     with gr.Tab("Step 2: Optimize Portfolio"):
         gr.HTML(STEP3_TITLE)
         with gr.Blocks() as app:
-            with gr.Row():
-                btn1 = gr.Button("Button 1")
-                btn2 = gr.Button("Button 2")
-                btn3 = gr.Button("Button 3")
-                btn4 = gr.Button("Button 4")
+            
             with gr.Row():
                 gr.HTML("<h1>Stock Portfolio Optimizer</h1>")
             
             with gr.Row():
                 start_date = gr.Textbox("2013-01-01", label="Start Date")
                 end_date = gr.Textbox(datetime.datetime.now().date(), label="End Date")
+            with gr.Row():
+                gr.HTML("<h1>Example Portfolios</h1>")
+                btn1 = gr.Button("Concervative")
+                btn2 = gr.Button("Growth")
+                btn3 = gr.Button("Balanced")
+                btn4 = gr.Button("Agressive")
+
+           
             
             with gr.Row():        
                 tickers_string = gr.Textbox("AAPL,AMZN,TSLA,NVDA,GOOGL,MSFT,NFLX,V,MA,CRM", 
                                             label='Enter all stock tickers to be included in portfolio separated \
                                             by commas WITHOUT spaces, e.g. "AAPL,AMZN,TSLA,NVDA,GOOGL,MSFT,NFLX,V,MA,CRM"')
+            with gr.Row(): 
                 btn = gr.Button("Get Optimized Portfolio")
                
             with gr.Row():
@@ -664,10 +674,28 @@ with gr.Blocks() as demo:
                 fig_indiv_prices = gr.Plot(label="Price of Individual Stocks")
                 fig_cum_returns = gr.Plot(label="Cumulative Returns of Individual Stocks Starting with $100")
         
-            btn.click(fn=output_results, inputs=[start_date, end_date, tickers_string], 
+            btn.click(fn=output_results, inputs=[start_date, end_date, ticker_string], 
                       outputs=[fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
                     expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns])
 
+            btn1.click(fn=output_results, inputs=[start_date, end_date, ticker_string_concervative], 
+                      outputs=[fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
+                    expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns])
+
+            btn2.click(fn=output_results, inputs=[start_date, end_date, ticker_string_growth], 
+                      outputs=[fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
+                    expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns])
+
+            btn3.click(fn=output_results, inputs=[start_date, end_date, ticker_string_balanced], 
+                      outputs=[fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
+                    expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns])
+
+            btn4.click(fn=output_results, inputs=[start_date, end_date, ticker_string_agressive], 
+                      outputs=[fig_cum_returns_optimized, weights_df, fig_efficient_frontier, fig_corr,   \
+                    expected_annual_return, annual_volatility, sharpe_ratio, fig_indiv_prices, fig_cum_returns])            
+
+    
+
     with gr.Tab("Step 3: Financial Outlook"): 
         gr.HTML(STEP2_TITLE)
         run_button_crewai = gr.Button(value="Run", variant="primary", scale=1)