Đặt biến MONGODB_DATABASE ở config và sửa nhẹ UI

app.py

@@ -23,7 +23,7 @@ def create_combined_app():
         
         with gr.Tabs() as tabs:
             # Tab 1: Vector Indexing (MongoDB to Qdrant)
-            with gr.Tab("🗄️ Vector Indexing", id="indexing"):
+            with gr.Tab("🗄️ Data Indexing", id="indexing"):
                 gr.Markdown("""
                 ## Indexing dữ liệu từ MongoDB lên Qdrant
                 
@@ -55,7 +55,7 @@ def create_combined_app():
         2. Cấu hình MongoDB connection string, database name và test connection
         3. Chọn loại sản phẩm hoặc xử lý tất cả. 
         
-        **Bước 2: Vector Indexing**
+        **Bước 2: Data Indexing**
         1. Chọn collection cần indexing
         2. Hệ thống sẽ tạo embeddings và đẩy lên Qdrant
         

config.py

@@ -26,6 +26,8 @@ IMAGE_EMBEDDING_MODEL = "google/efficientnet-b3"
 IMAGE_EMBEDDING_SIZE = 1536
 
 MONGODB_URI = os.getenv("MONGODB_URI", "mongodb://localhost:27017/")
+MONGODB_DATABASE = "dataset"
 
 QDRANT_HOST = os.getenv("QDRANT_HOST")
-QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
+QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
+

data_indexing.py

@@ -45,7 +45,8 @@ from config import (
     TEXT_EMBEDDING_MODEL,
     MONGODB_URI,
     QDRANT_HOST,
-    QDRANT_API_KEY
+    QDRANT_API_KEY,
+    MONGODB_DATABASE
 )
 
 from data_helper import *
@@ -154,7 +155,7 @@ mongodb_solution_collections = {
 
 """=================MONGODB CONNECTION========================"""
 class MongoDBConnection:
-    def __init__(self, connection_string: str = None, db_name: str = "product_database"):
+    def __init__(self, connection_string: str = None, db_name: str = MONGODB_DATABASE):
         """
         Initialize MongoDB connection
         Args:

data_processing.py

@@ -24,7 +24,7 @@ from pymongo import MongoClient, UpdateOne
 from pymongo.errors import ConnectionFailure, OperationFailure
 
 from data_helper import *
-from config import MONGODB_URI
+from config import MONGODB_URI, MONGODB_DATABASE
 
 # Suppress PyTorch DataLoader pin_memory warning on MPS
 warnings.filterwarnings("ignore", message=".*pin_memory.*not supported on MPS.*")
@@ -32,7 +32,7 @@ warnings.filterwarnings("ignore", message=".*pin_memory.*not supported on MPS.*"
 class MongoDBHandler:
     """Handler for MongoDB operations"""
     
-    def __init__(self, connection_string: str = None, database_name: str = "product_database"):
+    def __init__(self, connection_string: str = None, database_name: str = MONGODB_DATABASE):
         """
         Initialize MongoDB connection
         
@@ -512,7 +512,7 @@ def process_single_category(excel_path, category_name, processor_type,
         # Initialize MongoDB handler
         mongo_handler = MongoDBHandler(
             connection_string=mongo_connection if mongo_connection else None,
-            database_name=mongo_database if mongo_database else "product_database"
+            database_name=mongo_database if mongo_database else MONGODB_DATABASE
         )
         
         progress(0.3, desc=f"Processing {category_name} with {processor_type}...")
@@ -564,7 +564,7 @@ def test_mongo_connection(connection_string, database_name):
     if not connection_string:
         connection_string = "mongodb://localhost:27017/"
     if not database_name:
-        database_name = "product_database"
+        database_name = MONGODB_DATABASE
     
     handler = MongoDBHandler(connection_string, database_name)
     return handler.test_connection()
@@ -615,8 +615,8 @@ def create_processing_interface():
                 )
                 mongo_database = gr.Textbox(
                     label="Database Name",
-                    placeholder="product_database",
-                    value="product_database",
+                    placeholder="MONGODB_DATABASE",
+                    value=MONGODB_DATABASE,
                     info="Name of the MongoDB database"
                 )
                 test_connection_btn = gr.Button("🔌 Test Connection", size="sm")