app.py: prettify output

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import json
+
 from src.stock_analysis_agent import StockAnalyst
 
 
@@ -17,8 +19,26 @@ def interact_with_agent(agent_instance, ticker):
     if agent_instance is None:
         return 'Stock-Analysis Agent is not initialized. Please initialize first.'
     messages = agent_instance.get_stock_suggestion(ticker)
-    final_message = messages[3].model_dump()['content']
-    return final_message
+    FA_str = messages[-2].model_dump()['content']
+    summary_str = messages[-1].model_dump()['content']
+    response_pretty = ''
+    try:
+        FA_dict = json.loads(FA_str)
+        response_pretty += f"**Company Name:** {FA_dict['Company Name']}  \n"
+        response_pretty += f"**Sector:** {FA_dict['Sector']}\n\n"
+    except Exception as e:
+        response_pretty += f"**ticker**: {ticker}\n\n"
+        print(f'Error parsing the Financial Analysis response:\n{e}')
+    try:
+        summary_dict = json.loads(summary_str)
+        for key, value in summary_dict.items():
+            if key != 'stock':
+                pretty_key = key.replace('_', ' ').title()
+                response_pretty += f"**{pretty_key}**: {value}\n\n"
+    except Exception as e:
+        response_pretty += f'*An error occured stylizing the response, printing the raw response*:\n{summary_str}'
+        print(f'Error parsing summary response:\n{e}\n')
+    return response_pretty
 
 with gr.Blocks() as demo:
     gr.Markdown("# Stock Analysis Agent")
@@ -36,7 +56,7 @@ with gr.Blocks() as demo:
     gr.Markdown("Enter a stock symbol ('ticker') to be analyzed (GOOG, MSFT, etc.)")
     ticker = gr.Textbox(label="ticker")
     analyze_button = gr.Button('Analyze Stock')
-    output = gr.Textbox(label="Output Box")
+    output = gr.Markdown(label="Output Box")
     analyze_button.click(
             fn=interact_with_agent,
             inputs=[agent_instance, ticker],