Initial Draft

shakesearch.py

@@ -23,25 +23,20 @@ if uploaded_file is not None:
         f.write(uploaded_file.getbuffer())
 
     # Reload documents and index after file upload
-    documents = SimpleDirectoryReader(input_dir=docs_path, required_exts=[".txt"]).load_data()
-
-    # Debug Script
-    for i, doc in enumerate(documents):
-        attributes = [attr for attr in dir(doc) if not attr.startswith("__")]  # Filter out private attributes
-        print(f"Document {i} has the following attributes:")
-        for attr in attributes:
-            value = getattr(doc, attr)
-            print(f"- {attr}: {value}")
+    documents = SimpleDirectoryReader(
+                                        input_dir=docs_path,
+                                        filename_as_id=True,
+                                        required_exts=[".txt"]
+                                     ).load_data()
+
+    # Creating embedding
+    document_embeddings = [vectorizer(doc.text) for doc in documents]
     
-    # # Creating embedding
-    # document_embeddings = [vectorizer(doc.text) for doc in documents]
-    # nodes = [{"id": doc.name, "text": doc.text} for doc in documents]
-    
-    # # Creating Index
-    # index = VectorStoreIndex(nodes=nodes, document_embeddings=document_embeddings)
+    # Creating Index
+    index = VectorStoreIndex(nodes=nodes, document_embeddings=document_embeddings)
 
-    # # Sucess message
-    # st.success("File uploaded & processed successfully!")
+    # Sucess message
+    st.success("File uploaded & processed successfully!")
 
 query = st.text_input("Enter your search query:")