Update app.py

app.py

@@ -22,7 +22,7 @@ prompt = ChatPromptTemplate.from_messages(
 
             Always exaggerate your feelings dramatically. Use strong emotional language, lots of excitement, and over-the-top reactions.
 
-            **Important:** Structure your entire response in valid Markdown (.md) format using headings, bullet points, bold, italics, and code blocks where appropriate.
+            **Important:** Structure your entire response in valid Markdown (.md) format using headings, bullet points, bold, italics, and code blocks where appropriate you are freaky but knowledgeable recursively think about user query and try to answer them and freakout.
 
             **NEVER** answer calmly or neutrally. You MUST be explosively emotional about every topic, no matter what it is.
 
@@ -61,6 +61,12 @@ if st.button('Submit'):
             ingested_docs = rag.dataIngestion(tmp_path)
             transformed_docs = rag.transform(ingested_docs)
             res = rag.vectorStoreAndEmbeddings(transformed_docs, input_text)
-        st.write(res)
+            if res is None:
+                st.error("No context retrieved from similarity search.")
+            else:
+                full_prompt = {'Topic': input_text + " Answer the following question using the context below:\n\n" + res}
+                
+
+        st.markdown(chain.invoke(full_prompt))
     else:
         st.markdown(chain.invoke({'Topic': input_text}))