Update app.py

app.py

@@ -5,6 +5,7 @@ from langchain_community.vectorstores import FAISS
 from langchain_community.llms import CTransformers
 from langchain.chains import RetrievalQA
 import chainlit as cl
+import streamlit as st
 
 DB_FAISS_PATH = 'vectorstore/db_faiss'
 
@@ -91,4 +92,18 @@ async def main(message: cl.Message):
     else:
         answer += "\nNo sources found"
 
-    await cl.Message(content=answer).send()
+    await cl.Message(content=answer).send()
+
+
+def main():
+    st.title("Medical Bot")
+    st.text_input("Enter your query:", key="query")
+
+    if st.button("Submit"):
+        query = st.session_state['query']
+        with st.spinner("Thinking..."):
+            response = final_result(query)
+        st.markdown(response) 
+
+if __name__ == "__main__":
+    main()