Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 from langchain_community.document_loaders import PyPDFLoader, DirectoryLoader
 from langchain.prompts import PromptTemplate
 from langchain_community.embeddings import HuggingFaceEmbeddings
@@ -45,19 +45,20 @@ def qa_bot(query):
                                      chain_type_kwargs={'prompt': qa_prompt}
                                      )
     response = qa({'query': query})
-    return response
-
-def main():
-    st.title('Medical Bot')
-    query = st.text_input('Enter your medical query:')
-    if st.button('Submit'):
-        response = qa_bot(query)
-        st.write('Answer:', response['result'])
-        if response['source_documents']:
-            st.write('Sources:', response['source_documents'])
-        else:
-            st.write('No sources found')
-
-if __name__ == '__main__':
-    main()
+    return response['result'], response['source_documents']
+
+def main(query):
+    answer, sources = qa_bot(query)
+    if sources:
+        answer += f"\nSources: {sources}"
+    else:
+        answer += "\nNo sources found"
+    return answer
+
+iface = gr.Interface(fn=main, 
+                     inputs=gr.inputs.Textbox(label="Enter your medical query"),
+                     outputs=gr.outputs.Textbox(label="Answer"),
+                     title="Medical Bot",
+                     description="Ask any medical query and get an answer with sources if available.")
+iface.launch()