Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain_community.llms import HuggingFaceEndpoint
@@ -7,9 +7,9 @@ import docx2txt
 import io
 import re
 from typing import List
-from langchain.vectorstores import Chroma
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import SentenceTransformerEmbeddings
+from langchain_community.vectorstores import Chroma
+from langchain_community.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.embeddings import SentenceTransformerEmbeddings
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
@@ -132,20 +132,15 @@ def answer_query_with_similarity(query):
         print("An error occurred while getting the answer: ", str(e))
         return None
 
-def main():
-    st.title("Document Query App")
+def gradio_interface(query):
+    return answer_query_with_similarity(query)
 
-    query = st.text_input("Enter your query:")
-
-    if st.button("Get Answer"):
-        if query:
-            response = answer_query_with_similarity(query)
-            if response:
-                st.write("Answer:", response)
-            else:
-                st.write("No answer found.")
-        else:
-            st.write("Please provide a query.")
+interface = gr.Interface(
+    fn=gradio_interface,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your query here..."),
+    outputs="text",
+    title="Document Query App"
+)
 
 if __name__ == "__main__":
-    main()
+    interface.launch()