Update app.py

app.py

@@ -7,7 +7,7 @@ from typing import List
 from tempfile import NamedTemporaryFile
 from huggingface_hub import InferenceClient
 from langchain_community.document_loaders import PyPDFLoader
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain.docstore.document import Document
 
@@ -18,9 +18,14 @@ logging.basicConfig(level=logging.INFO)
 DOCUMENTS_FILE = "uploaded_documents.json"
 DEFAULT_MODEL = "@cf/meta/llama-2-7b-chat"
 HF_TOKEN = os.getenv("HF_API_TOKEN")  # Make sure to set this environment variable
+EMBED_MODEL = "sentence-transformers/all-mpnet-base-v2"
 
 def get_embeddings():
-    return HuggingFaceEmbeddings(model_name="avsolatorio/GIST-Embedding-v0")
+    return HuggingFaceEmbeddings(
+        model_name=EMBED_MODEL,
+        model_kwargs={'device': 'cpu'},
+        encode_kwargs={'normalize_embeddings': True}
+    )
 
 def load_documents():
     if os.path.exists(DOCUMENTS_FILE):
@@ -37,16 +42,17 @@ def load_document(file: NamedTemporaryFile) -> List[Document]:
     loader = PyPDFLoader(file.name)
     return loader.load_and_split()
 
-def update_vectors(files):
+def process_uploaded_files(files):
     if not files:
         return "Please upload at least one file.", []
     
+    files_list = [files] if not isinstance(files, list) else files
     embed = get_embeddings()
     uploaded_documents = load_documents()
     total_chunks = 0
     
     all_data = []
-    for file in files:
+    for file in files_list:
         try:
             data = load_document(file)
             if not data:
@@ -73,7 +79,7 @@ def update_vectors(files):
         database.save_local("faiss_database")
         
         save_documents(uploaded_documents)
-        return f"Vector store updated successfully. Processed {total_chunks} chunks.", uploaded_documents
+        return f"Vector store updated successfully. Processed {total_chunks} chunks.", [doc["name"] for doc in uploaded_documents]
         
     except Exception as e:
         return f"Error updating vector store: {str(e)}", []
@@ -102,7 +108,8 @@ def delete_documents(selected_docs):
         uploaded_documents = [doc for doc in uploaded_documents if doc["name"] not in selected_docs]
         save_documents(uploaded_documents)
         
-        return f"Deleted documents: {', '.join(selected_docs)}", uploaded_documents
+        remaining_docs = [doc["name"] for doc in uploaded_documents]
+        return f"Deleted documents: {', '.join(selected_docs)}", remaining_docs
     
     return "No documents to delete.", []
 
@@ -164,7 +171,7 @@ def create_interface():
                 files = gr.File(
                     label="Upload PDF Documents",
                     file_types=[".pdf"],
-                    multiple=True
+                    file_count="multiple"
                 )
                 upload_button = gr.Button("Upload and Process")
                 
@@ -179,22 +186,29 @@ def create_interface():
         
         with gr.Row():
             with gr.Column():
-                question = gr.Textbox(label="Ask a question about the documents")
+                question = gr.Textbox(
+                    label="Ask a question about the documents",
+                    placeholder="Enter your question here..."
+                )
                 temperature = gr.Slider(
                     minimum=0.0,
                     maximum=1.0,
                     value=0.2,
                     step=0.1,
-                    label="Temperature"
+                    label="Temperature (Higher values make the output more random)"
                 )
                 submit_button = gr.Button("Submit Question")
             
             with gr.Column():
-                answer = gr.Textbox(label="Answer", interactive=False)
+                answer = gr.Textbox(
+                    label="Answer",
+                    interactive=False,
+                    lines=10
+                )
         
         # Event handlers
         upload_button.click(
-            fn=update_vectors,
+            fn=process_uploaded_files,
             inputs=[files],
             outputs=[doc_status, doc_list]
         )
@@ -210,9 +224,20 @@ def create_interface():
             inputs=[question, temperature],
             outputs=[answer]
         )
+        
+        # Add keyboard shortcut for submitting questions
+        question.submit(
+            fn=get_response,
+            inputs=[question, temperature],
+            outputs=[answer]
+        )
     
     return app
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch(
+        server_name="0.0.0.0",  # Makes the app accessible from other machines
+        server_port=7860,       # Specify port
+        share=True             # Creates a public URL
+    )