Update app.py

app.py

@@ -19,6 +19,8 @@ from tradingview_ta import TA_Handler, Interval
 import yfinance as yf
 from datetime import datetime, timedelta
 from newsapi import NewsApiClient
+import json
+import altair as alt
 
 st.set_page_config(layout="wide")
 
@@ -211,21 +213,33 @@ if ticker_user!="":
     
     with col2:
         st.subheader("📝 Summary")
+        # Create bar chart with Altair
+        chart = alt.Chart(analysis_summary.summary).mark_bar().encode(
+            x=alt.X('RECOMMENDATION', sort=['BUY', 'NEUTRAL', 'SELL']),  # order if needed
+            y='Count',
+            color='RECOMMENDATION'
+        ).properties(
+            width=400,
+            height=300,
+            title="Recommendation Counts"
+        )
+        
+        st.altair_chart(chart, use_container_width=True)
         st.dataframe(analysis_summary.summary, use_container_width=True)
     
     # --- Row 2: Oscillators + Moving Averages ---
-    col3, col4 = st.columns(2)
-    with col3:
-        st.subheader("⚙️ Oscillator Analysis")
-        st.dataframe(analysis_summary.oscillators, use_container_width=True)
+    # col3, col4 = st.columns(2)
+    # with col3:
+    #     st.subheader("⚙️ Oscillator Analysis")
+    #     st.dataframe(analysis_summary.oscillators, use_container_width=True)
     
-    with col4:
-        st.subheader("📈 Moving Averages")
-        st.dataframe(analysis_summary.moving_averages, use_container_width=True)
+    # with col4:
+    #     st.subheader("📈 Moving Averages")
+    #     st.dataframe(analysis_summary.moving_averages, use_container_width=True)
     
-    # --- Row 3: Indicators ---
-    st.subheader("🔍 Indicators")
-    st.dataframe(analysis_summary.indicators, use_container_width=True)
+    # # --- Row 3: Indicators ---
+    # st.subheader("🔍 Indicators")
+    # st.dataframe(analysis_summary.indicators, use_container_width=True)
 
     url = "https://api.chart-img.com/v2/tradingview/advanced-chart"
     api_key = "l0iUFRSeqC9z7nDPTd1hnafPh2RrdcEy6rl6tNqV"
@@ -254,7 +268,6 @@ if ticker_user!="":
         llm_chain = LLMChain(llm=llm,prompt=llm_prompt)
         stuff_chain = StuffDocumentsChain(llm_chain=llm_chain,document_variable_name="input_documents")
         
-        st.write(stuff_chain.input_keys) 
         # res = stuff_chain.invoke(docs)
         res = stuff_chain.invoke({"input_documents": docs})
         #create the humanmassage propmt templete with the image file 
@@ -271,7 +284,7 @@ if ticker_user!="":
         
         # st.write(message.content)
         # st.write(res["output_text"])
-        data = res["output_text"]
+        data = json.loads(res["output_text"])
         # Header Info
         st.markdown(f"### {data['stock_summary']['company_name']} ({data['stock_summary']['ticker']}) | {data['stock_summary']['exchange']}")
         st.markdown(f"**Description**: {data['stock_summary']['description']}")