Update app.py

app.py

@@ -67,46 +67,46 @@ def stream_generator(prompt, thread_id):
         except Exception as e:
             print(f"Error during streaming: {str(e)}")
 
-# def upload_and_add_to_vector_store(uploaded_file):
-#     """Upload a file to OpenAI and add it to the specified vector store."""
-#     try:
-#         # Convert the uploaded file to a BytesIO stream for uploading
-#         file_stream = io.BytesIO(uploaded_file.getvalue())
-#         file_stream.name = uploaded_file.name  # Preserve the file name    
-#         # Upload the file to the vector store
-#         file_batch = client.beta.vector_stores.file_batches.upload_and_poll(
-#             vector_store_id=vector_store_id,
-#             files=[file_stream]
-#         )
-
-
-#         st.success(f"File '{uploaded_file.name}' processed and added to vector store. Status: {file_batch.status}")
-#     except Exception as e:
-#         st.error(f"Failed to process file: {str(e)}")
+def upload_and_add_to_vector_store(uploaded_file):
+    """Upload a file to OpenAI and add it to the specified vector store."""
+    try:
+        # Convert the uploaded file to a BytesIO stream for uploading
+        file_stream = io.BytesIO(uploaded_file.getvalue())
+        file_stream.name = uploaded_file.name  # Preserve the file name    
+        # Upload the file to the vector store
+        file_batch = client.beta.vector_stores.file_batches.upload_and_poll(
+            vector_store_id=vector_store_id,
+            files=[file_stream]
+        )
+
+
+        st.success(f"File '{uploaded_file.name}' processed and added to vector store. Status: {file_batch.status}")
+    except Exception as e:
+        st.error(f"Failed to process file: {str(e)}")
     
 
-# def list_all_files_in_vector_store():
-#     """List all files in the specified vector store."""
-#     try:
-#         all_files = list(client.beta.vector_stores.files.list(vector_store_id=vector_store_id))
-#         # st.write(all_files)
-#         for file in all_files:
-#             file_id = file.id
-#             st.write(file_id)   
-#     except Exception as e:
-#         st.error(f"Failed to list files: {str(e)}")
-#         return {}
-
-# def delete_file_from_vector_store(vector_store_id, file_id):
-#     """Delete a file from the specified vector store."""
-#     try:
-#         client.beta.vector_stores.files.delete(
-#             vector_store_id=vector_store_id,
-#             file_id=file_id
-#         )
-#         st.success(f"File with ID '{file_id}' deleted from vector store '{vector_store_id}'.")
-#     except Exception as e:
-#         st.error(f"Failed to delete file. File id is not Found.")
+def list_all_files_in_vector_store():
+    """List all files in the specified vector store."""
+    try:
+        all_files = list(client.beta.vector_stores.files.list(vector_store_id=vector_store_id))
+        # st.write(all_files)
+        for file in all_files:
+            file_id = file.id
+            st.write(file_id)   
+    except Exception as e:
+        st.error(f"Failed to list files: {str(e)}")
+        return {}
+
+def delete_file_from_vector_store(vector_store_id, file_id):
+    """Delete a file from the specified vector store."""
+    try:
+        client.beta.vector_stores.files.delete(
+            vector_store_id=vector_store_id,
+            file_id=file_id
+        )
+        st.success(f"File with ID '{file_id}' deleted from vector store '{vector_store_id}'.")
+    except Exception as e:
+        st.error(f"Failed to delete file. File id is not Found.")
 
 # Interface to delete files from vector store
 st.sidebar.subheader("Delete File from Vector Store")