Update app.py

app.py

@@ -94,12 +94,14 @@ def index():
                     logger.info(f"Deleted ChromaDB collection: {DB_NAME}")
                 except Exception as e:
                     logger.warning(f"Failed to delete collection {DB_NAME}: {e}")
-                client.create_collection(DB_NAME)
+                collection = client.create_collection(DB_NAME)
                 logger.info(f"Created fresh ChromaDB collection: {DB_NAME}")
                 
                 # Process dataset
-                result = subprocess.run(['python', 'process_hf_dataset.py'], check=True, capture_output=True, text=True)
+                result = subprocess.run(['python', 'process_hf_dataset.py'], check=True, capture_output=True, text=True, cwd=os.path.dirname(__file__))
                 logger.info(f"Process Hugging Face dataset output: {result.stdout}")
+                if result.stderr:
+                    logger.error(f"Process Hugging Face dataset errors: {result.stderr}")
                 return render_template(
                     'results_partial.html',
                     parts=None,
@@ -121,8 +123,10 @@ def index():
                 logger.info(f"Using existing or new ChromaDB collection: {DB_NAME}")
                 
                 # Process dataset
-                result = subprocess.run(['python', 'process_hf_dataset.py'], check=True, capture_output=True, text=True)
+                result = subprocess.run(['python', 'process_hf_dataset.py'], check=True, capture_output=True, text=True, cwd=os.path.dirname(__file__))
                 logger.info(f"Load Hugging Face dataset output: {result.stdout}")
+                if result.stderr:
+                    logger.error(f"Load Hugging Face dataset errors: {result.stderr}")
                 return render_template(
                     'results_partial.html',
                     parts=None,
@@ -144,8 +148,11 @@ def index():
                     logger.info(f"Deleted ChromaDB collection: {DB_NAME}")
                 except Exception as e:
                     logger.warning(f"Failed to delete collection {DB_NAME}: {e}")
-                client.create_collection(DB_NAME)
+                collection = client.create_collection(DB_NAME)
                 logger.info(f"Created fresh ChromaDB collection: {DB_NAME}")
+                # Verify collection creation by checking if it's iterable (fix for 'NoneType' error)
+                if collection is None or not hasattr(collection, 'add'):
+                    raise ValueError("ChromaDB collection creation failed")
                 return render_template(
                     'results_partial.html',
                     parts=None,