Update app.py

app.py

@@ -4,9 +4,8 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
 import pickle
-from langchain_community.llms import HuggingFaceEndpoint
-from langchain_community.vectorstores import FAISS
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface.llms import HuggingFaceEndpoint
+from langchain_huggingface.embeddings import HuggingFaceEmbeddings
 import gradio as gr
 
 # Initialize the embedding model
@@ -110,19 +109,25 @@ def query_text(text):
         print(f"Error querying text: {e}")
         return f"Error querying text: {e}"
 
-# Sample Gradio integration (for illustration)
 def main():
-    gr.Interface(
-        [
-            upload_files,
-            query_text
-        ],
-        [
-            "files",
-            "text"
-        ],
-        "text"
-    ).launch()
+    upload_interface = gr.Interface(
+        fn=upload_files,
+        inputs=gr.inputs.File(file_count="multiple", label="Upload DOCX files"),
+        outputs="text",
+        title="Upload DOCX Files",
+        description="Upload DOCX files to process and add to the FAISS index."
+    )
+
+    query_interface = gr.Interface(
+        fn=query_text,
+        inputs="text",
+        outputs="text",
+        title="Query Text",
+        description="Query the indexed text and get answers from the language model."
+    )
+
+    demo = gr.TabbedInterface([upload_interface, query_interface], ["Upload Files", "Query Text"])
+    demo.launch()
 
 if __name__ == "__main__":
     main()
@@ -136,3 +141,4 @@ if __name__ == "__main__":
 
 
 
+