Update app.py

app.py

@@ -15,12 +15,18 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains.llm import LLMChain
 from langchain.chains import StuffDocumentsChain
 from langchain_core.messages import HumanMessage
+import requests
+st.set_page_config(layout="wide")
 
 GOOGLE_API_KEY=os.environ['GOOGLE_API_KEY']
 
 st.title('Stock Market Insights')
+st.subheader('Using [](https://www.gstatic.com/lamda/images/gemini_wordmark_landing_page_238102af073d0ae2763aa5.svg)')
 st.sidebar.info("Know more about [NSE Tickers](https://www.google.com/search?q=nse+tickers+list&sca_esv=a6c39f4d03c5324c&sca_upv=1&rlz=1C1GCEB_enIN1011IN1011&sxsrf=ADLYWILQPbew-0SrvUUWpI8Y29_uOOgbvA%3A1716470016765&ei=AEFPZp-zLvzHp84P_ZWtuA0&oq=NSE+Tickers+&gs_lp=Egxnd3Mtd2l6LXNlcnAiDE5TRSBUaWNrZXJzICoCCAAyBRAAGIAEMggQABgWGAoYHjIGEAAYFhgeMgYQABgWGB4yBhAAGBYYHjIGEAAYFhgeMgYQABgWGB4yBhAAGBYYHjILEAAYgAQYhgMYigUyCxAAGIAEGIYDGIoFSIIbUL0PWL0PcAF4AZABAJgB8QKgAfECqgEDMy0xuAEByAEA-AEBmAICoAKKA8ICChAAGLADGNYEGEeYAwCIBgGQBgiSBwUxLjMtMaAHtQU&sclient=gws-wiz-serp)")
-    
+
+st.sidebar.info("Know more about [Charts](https://chart-img.com/)")
+st.sidebar.markdown("The App uses Gemini API for Text and Vision along with LangChain")
+st.image("https://myndroot.com/wp-content/uploads/2023/12/Gemini-Dext.jpg")
 ticker_user = st.text_input("Enter Ticker for NSE Stocks","")
     
 gemini_embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
@@ -59,17 +65,7 @@ if ticker_user!="":
     """
     
     st.sidebar.subheader('Prompt')
-    user_prompt = st.sidebar.text_area("Enter Prompt",llm_prompt_template)
-    
-    llm_prompt = PromptTemplate.from_template(user_prompt)
-    
-    llm_chain = LLMChain(llm=llm,prompt=llm_prompt)
-    stuff_chain = StuffDocumentsChain(llm_chain=llm_chain,document_variable_name="context")
-    
-    
-    res = stuff_chain.invoke(docs)
-
-    import requests
+    # user_prompt = st.sidebar.text_area("Enter Prompt",llm_prompt_template)
     
     url = "https://api.chart-img.com/v2/tradingview/advanced-chart"
     api_key = "l0iUFRSeqC9z7nDPTd1hnafPh2RrdcEy6rl6tNqV"
@@ -92,25 +88,36 @@ if ticker_user!="":
             f.write(response.content)
             
         st.image("chart_t1.jpg", caption='')
+
+        llm_prompt = PromptTemplate.from_template(llm_prompt_template)
+    
+        llm_chain = LLMChain(llm=llm,prompt=llm_prompt)
+        stuff_chain = StuffDocumentsChain(llm_chain=llm_chain,document_variable_name="context")
+        
+        
+        res = stuff_chain.invoke(docs)
+
+        #create the humanmassage propmt templete with the image file 
+        hmessage = HumanMessage(
+            content=[
+                {
+                    "type": "text",
+                    "text": "Based on the Image, suggest a BUY and SELL Strategy along with Risk based approach using Stop loss/Target price",
+                },
+                {"type": "image_url", "image_url": "chart_t1.jpg"},
+            ]
+        )
+        message = llm_vis.invoke([hmessage])
+        
+        st.write(message.content)
+        st.write(res["output_text"])
+    
         # print("Image saved as chart-img-02.png")
     else:
-        st.write(f"Failed to retrieve image. Status code: {response.status_code}")
-        st.write("Response:", response.text)
-
-    #create the humanmassage propmt templete with the image file 
-    hmessage = HumanMessage(
-        content=[
-            {
-                "type": "text",
-                "text": "Based on the chart, could you predict the movement and suggest a BUY and SELL Strategy",
-            },
-            {"type": "image_url", "image_url": "chart_t1.jpg"},
-        ]
-    )
-    message = llm_vis.invoke([hmessage])
-    
-    st.write(message.content)
-    st.write(res["output_text"])
+        st.warning(f"Failed to retrieve image. Status code: {response.status_code}")
+        st.warning("Response:", response.text)
+
+