Update app.py

app.py

@@ -11,10 +11,8 @@ from urllib.parse import quote_plus
 user = quote_plus(os.getenv("MONGO_USER"))
 password = quote_plus(os.getenv("MONGO_PASS"))
 cluster = os.getenv("MONGO_CLUSTER")
-# db_name = "grant_docs"
 db_name = os.environ.get("MONGO_DB", "grant_docs")
 mongo_uri = f"mongodb+srv://{user}:{password}@{cluster}/{db_name}?retryWrites=true&w=majority&tls=true&tlsAllowInvalidCertificates=true"
-# mongo_uri = os.environ["MONGO_URI"]
 client = MongoClient(mongo_uri, tls=True, tlsAllowInvalidCertificates=True, serverSelectionTimeoutMS=20000)
 db = client[db_name]
 
@@ -33,17 +31,17 @@ with st.sidebar:
     except Exception as e:
         st.error(f"Failed to list collections: {e}")
         selected_collection = "doc_chunks_cat"
+
     if selected_collection == "Create New Collection":
-        selected_collection = st.sidebar.text_input("Enter Collection Name:)")
+        selected_collection = st.sidebar.text_input("Enter Collection Name:")
         if not selected_collection:
             st.warning("⚠️ Enter a collection name to proceed.")
             st.stop()
+
     is_grant_app = st.toggle("Is this a Grant Application?", value=False)
 
-# uploaded_file = st.file_uploader("Upload a DOCX or TXT file", type=["docx", "txt"])
 uploaded_file = st.file_uploader("Upload a DOCX, TXT, or PDF file", type=["docx", "txt", "pdf"])
 
-
 if uploaded_file:
     temp_path = Path(tempfile.gettempdir()) / uploaded_file.name
     with open(temp_path, "wb") as f:
@@ -69,10 +67,9 @@ if uploaded_file:
                     "is_grant_app": is_grant_app,
                 })
                 collection.insert_one(chunk)
-            
+
             st.success(f"✅ {len(chunks)} chunks inserted into `{selected_collection}`")
 
-            # Show a few previews
             for i, c in enumerate(chunks[:3]):
                 st.subheader(f"Chunk {i+1}: {c['metadata'].get('header') or 'No Header'}")
                 st.markdown(c['text'][:400] + "...")
@@ -81,13 +78,14 @@ if uploaded_file:
 
             if len(chunks) > 3:
                 st.info(f"... and {len(chunks)-3} more chunks processed.")
-                # st.success(f"✅ {len(chunks)} chunks inserted into `{selected_collection}`")
-                try:
-                    os.remove(temp_path)
-                except Exception as e:
-                    st.warning(f"⚠️ Could not delete temp file: {e}")
-                # Rerun the app to refresh the UI
-                st.rerun()
 
         else:
             st.warning("⚠️ No chunks were generated.")
+
+        try:
+            if temp_path.exists():
+                os.remove(temp_path)
+        except Exception as e:
+            st.warning(f"⚠️ Could not delete temp file: {e}")
+
+        st.rerun()