Update app.py

app.py

@@ -48,7 +48,9 @@ if not GOOGLE_AI_STUDIO:
     raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
 
 
-
+STEP1_TITLE = """<h1 align="center">Describe Your Investment Goals & Create a Portfolio</h1>"""
+STEP2_TITLE = """<h1 align="center">Create a Financial OutLook Report of Your Portfolio</h1>"""
+STEP3_TITLE = """<h1 align="center">Optimize Your Portfolio to Maximize Return</h1>"""
 
 
 # LangChain function for company analysis
@@ -93,15 +95,12 @@ def company_analysis(api_key: str, company_name: str) -> dict:
 
 
 
-
-
-
-
 print("google-generativeai:", genai.__version__)
 
+
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
-TITLE_INTRO = """<h1 align="center">Introduction to Financial Manager</h1>"""
+
 
 TITLE1 = """<h1 align="center">Company Analysis</h1>"""
 TITLE2 = """<h1 align="center">Investment Strategy</h1>"""
@@ -501,10 +500,13 @@ def output_results(start_date, end_date, tickers_string):
     
 with gr.Blocks() as demo:
 
-    with gr.Tab("Introduction"):
-        gr.HTML(TITLE_INTRO)
+    with gr.Tab("Step 1: Investment Goals"):
+        gr.HTML(STEP1_TITLE)
+
+        
     
-    with gr.Tab("Your Portfolio"): 
+    with gr.Tab("Step 2: Financial Outlook"): 
+        gr.HTML(STEP2_TITLE)
         run_button_crewai = gr.Button(value="Run", variant="primary", scale=1)
         run_button_crewai.click(
             fn=crewai_process,
@@ -513,10 +515,11 @@ with gr.Blocks() as demo:
         )
             
         
-    with gr.Tab("Portfolio Analysis"):
+    with gr.Tab("Step 3: Optimize Portfolio"):
+        gr.HTML(STEP3_TITLE)
         with gr.Blocks() as app:
             with gr.Row():
-                gr.HTML("<h1>Bohmian's Stock Portfolio Optimizer</h1>")
+                gr.HTML("<h1>Stock Portfolio Optimizer</h1>")
             
             with gr.Row():
                 start_date = gr.Textbox("2013-01-01", label="Start Date")