"Update space"

Signed-off-by: Emmanuel Frimpong Asante <[email protected]>

app.py

@@ -5,10 +5,8 @@ import tensorflow as tf
 from pymongo import MongoClient
 from flask import Flask, request, jsonify, render_template
 from huggingface_hub import login
-import threading
 from services.disease_detection import PoultryFarmBot
 from services.llama_service import llama2_response
-from config.db import db
 
 # Initialize Flask application
 app = Flask(__name__, template_folder="templates", static_folder="static")

data/logs/daily_logs.py

@@ -3,9 +3,7 @@
 from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (import from your db configuration)
-from config.db import db
+from config.db import db  # MongoDB connection from config
 
 # Define the daily logs collection
 daily_logs_collection = db.daily_logs  # Collection to store daily logs
@@ -134,26 +132,3 @@ def delete_daily_log(log_id):
     """
     result = daily_logs_collection.delete_one({"_id": ObjectId(log_id)})
     return result.deleted_count > 0
-
-
-# Example function to create a sample daily log
-if __name__ == "__main__":
-    # Insert a test daily log
-    test_user_id = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-    test_date = "2024-10-07"
-    test_health_status = "All poultry are healthy, no signs of disease."
-    test_feed_consumption = 500  # In kilograms
-    test_mortality_rate = 0.0  # No deaths today
-    test_inventory_levels = {"Feed": 1000, "Antibiotics": 50}  # Inventory in kilograms and units
-
-    log_id = create_daily_log(
-        user_id=test_user_id,
-        date=test_date,
-        health_status=test_health_status,
-        feed_consumption=test_feed_consumption,
-        mortality_rate=test_mortality_rate,
-        inventory_levels=test_inventory_levels,
-        notes="Everything ran smoothly today."
-    )
-
-    print(f"Daily log created with ID: {log_id}")

data/logs/user_usage.py

@@ -3,9 +3,7 @@
 from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (import from your db configuration)
-from config.db import db
+from config.db import db  # MongoDB connection from configuration
 
 # Define the user usage logs collection
 user_usage_collection = db.user_usage  # Collection to store assistant usage logs
@@ -113,21 +111,3 @@ def delete_usage_log(log_id):
     """
     result = user_usage_collection.delete_one({"_id": ObjectId(log_id)})
     return result.deleted_count > 0
-
-
-# Example function to create a sample user usage log
-if __name__ == "__main__":
-    # Insert a test user usage log
-    test_user_id = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-    test_manager_id = "610c5e8a85f0b2a4e0b8b919"  # Replace with a valid manager ObjectId
-    test_interaction_type = "disease_detection"
-    test_metadata = {"disease_name": "Coccidiosis", "detected": True}
-
-    log_id = create_user_usage_log(
-        user_id=test_user_id,
-        manager_id=test_manager_id,
-        interaction_type=test_interaction_type,
-        metadata=test_metadata
-    )
-
-    print(f"User usage log created with ID: {log_id}")

models/schemas/health_schema.py

@@ -1,11 +1,8 @@
 # models/schemas/health_schema.py
 
-from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (move to config/db.py and import here)
-from config.db import db
+from config.db import db  # MongoDB connection from configuration
 
 # Define the health collection
 health_collection = db.health  # Collection to store poultry health records
@@ -118,23 +115,3 @@ def delete_health_record(record_id):
     """
     result = health_collection.delete_one({"_id": ObjectId(record_id)})
     return result.deleted_count > 0
-
-
-# Example function to create a sample health record
-if __name__ == "__main__":
-    # Insert a test health record
-    test_user_id = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-    test_poultry_id = "poultry123"
-    test_disease_name = "Coccidiosis"
-    test_status = "Critical"
-    test_recommendation = "Panadol"
-
-    record_id = create_health_record(
-        user_id=test_user_id,
-        poultry_id=test_poultry_id,
-        disease_name=test_disease_name,
-        status=test_status,
-        recommendation=test_recommendation
-    )
-
-    print(f"Health record created with ID: {record_id}")

models/schemas/inventory_schema.py

@@ -1,12 +1,9 @@
 # models/schemas/inventory_schema.py
 
 
-from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (move to config/db.py and import here)
-from config.db import db
+from config.db import db  # MongoDB connection from configuration
 
 # Define the inventory collection
 inventory_collection = db.inventory  # Collection to store inventory items
@@ -126,24 +123,3 @@ def delete_inventory_item(item_id):
     """
     result = inventory_collection.delete_one({"_id": ObjectId(item_id)})
     return result.deleted_count > 0
-
-
-# Example function to create a sample inventory item
-if __name__ == "__main__":
-    # Insert a test inventory item
-    test_item_name = "Chicken Feed"
-    test_category = "Feed"
-    test_quantity = 1000  # 1000 kg
-    test_unit = "kg"
-    test_added_by = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-
-    item_id = create_inventory_item(
-        item_name=test_item_name,
-        category=test_category,
-        quantity=test_quantity,
-        unit=test_unit,
-        added_by=test_added_by,
-        description="Premium feed for broiler chickens"
-    )
-
-    print(f"Inventory item created with ID: {item_id}")

models/schemas/log_schema.py

@@ -1,17 +1,15 @@
 # models/schemas/log_schema.py
 
 
-from pymongo import MongoClient, IndexModel, ASCENDING
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (move to config/db.py and import here)
-from config.db import db
+from pymongo import ASCENDING
+from config.db import db  # Import MongoDB connection from config
 
 # Define the logs collection
 logs_collection = db.logs  # Collection to store system logs
 
-# TTL index to expire documents after 30 days
+# Create a TTL index to expire documents after 30 days
 logs_collection.create_index([("timestamp", ASCENDING)], expireAfterSeconds=30 * 24 * 60 * 60)
 
 
@@ -110,23 +108,3 @@ def delete_old_logs():
     """
     # No manual deletion needed as MongoDB TTL will automatically handle the deletion.
     pass
-
-
-# Example function to create a sample log
-if __name__ == "__main__":
-    # Insert a test log record
-    test_level = "ERROR"
-    test_message = "An error occurred in the system."
-    test_metadata = {"stack_trace": "Traceback (most recent call last): ..."}
-    test_ip_address = "192.168.1.1"
-    test_user_id = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-
-    log_id = create_log(
-        level=test_level,
-        message=test_message,
-        metadata=test_metadata,
-        ip_address=test_ip_address,
-        user_id=test_user_id
-    )
-
-    print(f"Log record created with ID: {log_id}")

models/schemas/usage_schema.py

@@ -1,12 +1,9 @@
 # models/schemas/usage_schema.py
 
 
-from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB connection (move to config/db.py and import here)
-from config.db import db
+from config.db import db  # Import MongoDB connection from config
 
 # Define the usage collection
 usage_collection = db.usage  # Collection to store assistant usage records
@@ -104,21 +101,3 @@ def delete_usage_record(record_id):
     """
     result = usage_collection.delete_one({"_id": ObjectId(record_id)})
     return result.deleted_count > 0
-
-
-# Example function to create a sample usage record
-if __name__ == "__main__":
-    # Insert a test usage record
-    test_user_id = "610c5e8a85f0b2a4e0b8b918"  # Replace with a valid user ObjectId
-    test_manager_id = "610c5e8a85f0b2a4e0b8b919"  # Replace with a valid manager ObjectId
-    test_interaction_type = "disease_detection"
-    test_metadata = {"disease_name": "Coccidiosis", "detected": True}
-
-    record_id = create_usage_record(
-        user_id=test_user_id,
-        manager_id=test_manager_id,
-        interaction_type=test_interaction_type,
-        metadata=test_metadata
-    )
-
-    print(f"Usage record created with ID: {record_id}")

routes/api.py

@@ -10,153 +10,58 @@ from controllers.data_logging import log_user_action
 # Create a Blueprint for the API routes
 api_bp = Blueprint('api', __name__)
 
-
 # Health Management Routes
 @api_bp.route('/detect-disease', methods=['POST'])
 def detect_disease_route():
-    """
-    Route for detecting a disease and logging the health record.
-    Expects a POST request with:
-    - user_id: ID of the user.
-    - poultry_id: ID of the poultry.
-    - image: The image of poultry feces for disease detection.
-
-    Returns: JSON response with disease detection results and log status.
-    """
     return detect_and_log_disease()
 
-
 @api_bp.route('/health-record/<record_id>', methods=['GET'])
 def get_health_record_route(record_id):
-    """
-    Route for retrieving a health record by its ID.
-
-    Returns: JSON response with the health record details.
-    """
     return get_health_record(record_id)
 
-
 @api_bp.route('/health-records/<user_id>', methods=['GET'])
 def get_health_records_for_user_route(user_id):
-    """
-    Route for retrieving all health records for a specific user.
-
-    Returns: JSON response with the list of health records.
-    """
     return get_health_records_for_user(user_id)
 
-
 # Inventory Management Routes
 @api_bp.route('/add-inventory-item', methods=['POST'])
 def add_inventory_item_route():
-    """
-    Route for adding a new inventory item.
-    Expects a POST request with:
-    - user_id: ID of the user.
-    - item_name: Name of the item.
-    - category: Category of the item.
-    - quantity: Quantity of the item.
-    - unit: Unit of measurement.
-
-    Returns: JSON response with the item addition status.
-    """
     return add_inventory_item()
 
-
 @api_bp.route('/update-inventory-item', methods=['PUT'])
 def update_inventory_item_route():
-    """
-    Route for updating an inventory item.
-    Expects a PUT request with:
-    - user_id: ID of the user.
-    - item_id: ID of the inventory item to update.
-    - updated_fields: Fields to update (e.g., quantity, used_quantity).
-
-    Returns: JSON response with the item update status.
-    """
     return update_inventory_item_by_id()
 
-
 @api_bp.route('/inventory-item/<item_id>', methods=['GET'])
 def get_inventory_item_route(item_id):
-    """
-    Route for retrieving an inventory item by its ID.
-
-    Returns: JSON response with the inventory item details.
-    """
     return get_inventory_by_id(item_id)
 
-
 @api_bp.route('/inventory', methods=['GET'])
 def get_all_inventory_route():
-    """
-    Route for retrieving all inventory items.
-
-    Returns: JSON response with the list of all inventory items.
-    """
     return get_all_inventory()
 
-
 # Assistant Usage Tracking Routes
 @api_bp.route('/log-usage', methods=['POST'])
 def log_usage_route():
-    """
-    Route for logging the usage of the assistant by a user.
-    Expects a POST request with:
-    - user_id: ID of the user.
-    - manager_id: ID of the manager.
-    - interaction_type: The type of interaction.
-
-    Returns: JSON response with the log status.
-    """
     return log_usage()
 
-
 @api_bp.route('/usage/<user_id>', methods=['GET'])
 def get_usage_for_user_route(user_id):
-    """
-    Route for retrieving all usage logs for a specific user.
-
-    Returns: JSON response with the list of usage logs for the user.
-    """
     return get_usage_for_user(user_id)
 
-
 @api_bp.route('/manager-usage/<manager_id>', methods=['GET'])
 def get_usage_for_manager_route(manager_id):
-    """
-    Route for retrieving all usage logs for users managed by a specific manager.
-
-    Returns: JSON response with the list of usage logs for the manager's users.
-    """
     return get_usage_for_manager(manager_id)
 
-
 @api_bp.route('/usage-type/<interaction_type>', methods=['GET'])
 def get_usage_by_type_route(interaction_type):
-    """
-    Route for retrieving all usage logs for a specific interaction type.
-
-    Returns: JSON response with the list of usage logs for the interaction type.
-    """
     return get_usage_by_type(interaction_type)
 
-
 # User Action Logging Route
 @api_bp.route('/log-user-action', methods=['POST'])
 def log_user_action_route():
-    """
-    Route for logging general user actions.
-    Expects a POST request with:
-    - user_id: ID of the user performing the action.
-    - message: Description of the action.
-    - level: Logging level (default is "INFO").
-
-    Returns: JSON response with the log status.
-    """
     user_id = request.json.get('user_id')
     message = request.json.get('message')
     level = request.json.get('level', 'INFO')
     metadata = request.json.get('metadata', {})
-
     return log_user_action(user_id, message, level, metadata)

routes/data_logging_routes.py

@@ -2,27 +2,20 @@
 
 
 from flask import Blueprint, request, jsonify
-from controllers.data_logging import log_user_action, log_health_record, log_inventory_addition, log_inventory_update, \
+from controllers.data_logging import (
+    log_user_action,
+    log_health_record,
+    log_inventory_addition,
+    log_inventory_update,
     log_assistant_usage
+)
 
 # Create a Blueprint for the data logging routes
 data_logging_bp = Blueprint('data_logging', __name__)
 
-
 # Route to log user actions
 @data_logging_bp.route('/log-user-action', methods=['POST'])
 def log_user_action_route():
-    """
-    Logs a general user action.
-
-    Expects a POST request with:
-    - user_id: ID of the user performing the action.
-    - message: Description of the action.
-    - level: Logging level (default is "INFO").
-    - metadata: Optional. Additional metadata related to the action.
-
-    Returns: JSON response indicating success or failure.
-    """
     try:
         user_id = request.json.get('user_id')
         message = request.json.get('message')
@@ -33,7 +26,6 @@ def log_user_action_route():
             return jsonify({"error": "Missing required fields"}), 400
 
         return log_user_action(user_id, message, level, metadata)
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -41,19 +33,6 @@ def log_user_action_route():
 # Route to log a health record
 @data_logging_bp.route('/log-health-record', methods=['POST'])
 def log_health_record_route():
-    """
-    Logs a health record for poultry.
-
-    Expects a POST request with:
-    - user_id: ID of the user submitting the record.
-    - poultry_id: The ID of the poultry.
-    - disease_name: Name of the detected disease.
-    - status: The health status of the poultry.
-    - recommendation: Recommended treatment.
-    - image_path: Optional. Path to an image related to the health record.
-
-    Returns: JSON response indicating success or failure.
-    """
     try:
         user_id = request.json.get('user_id')
         poultry_id = request.json.get('poultry_id')
@@ -62,11 +41,10 @@ def log_health_record_route():
         recommendation = request.json.get('recommendation')
         image_path = request.json.get('image_path')
 
-        if not user_id or not poultry_id or not disease_name or not status or not recommendation:
+        if not all([user_id, poultry_id, disease_name, status, recommendation]):
             return jsonify({"error": "Missing required fields"}), 400
 
         return log_health_record(user_id, poultry_id, disease_name, status, recommendation, image_path)
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -74,19 +52,6 @@ def log_health_record_route():
 # Route to log inventory addition
 @data_logging_bp.route('/log-inventory-addition', methods=['POST'])
 def log_inventory_addition_route():
-    """
-    Logs the addition of an inventory item.
-
-    Expects a POST request with:
-    - user_id: ID of the user adding the item.
-    - item_name: Name of the inventory item.
-    - category: Category of the item (e.g., "Feed", "Medicine").
-    - quantity: Total quantity added.
-    - unit: Unit of measurement (e.g., "kg", "liters").
-    - description: Optional. Additional description or notes.
-
-    Returns: JSON response indicating success or failure.
-    """
     try:
         user_id = request.json.get('user_id')
         item_name = request.json.get('item_name')
@@ -95,11 +60,10 @@ def log_inventory_addition_route():
         unit = request.json.get('unit')
         description = request.json.get('description')
 
-        if not user_id or not item_name or not category or not quantity or not unit:
+        if not all([user_id, item_name, category, quantity, unit]):
             return jsonify({"error": "Missing required fields"}), 400
 
         return log_inventory_addition(user_id, item_name, category, quantity, unit, description)
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -107,26 +71,15 @@ def log_inventory_addition_route():
 # Route to log inventory update
 @data_logging_bp.route('/log-inventory-update', methods=['PUT'])
 def log_inventory_update_route():
-    """
-    Logs an update to an existing inventory item.
-
-    Expects a PUT request with:
-    - user_id: ID of the user updating the item.
-    - item_id: ID of the inventory item being updated.
-    - updated_fields: Fields being updated (e.g., "quantity", "used_quantity").
-
-    Returns: JSON response indicating success or failure.
-    """
     try:
         user_id = request.json.get('user_id')
         item_id = request.json.get('item_id')
         updated_fields = request.json.get('updated_fields')
 
-        if not user_id or not item_id or not updated_fields:
+        if not all([user_id, item_id, updated_fields]):
             return jsonify({"error": "Missing required fields"}), 400
 
         return log_inventory_update(user_id, item_id, updated_fields)
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -134,27 +87,15 @@ def log_inventory_update_route():
 # Route to log assistant usage
 @data_logging_bp.route('/log-assistant-usage', methods=['POST'])
 def log_assistant_usage_route():
-    """
-    Logs the usage of the assistant by a user.
-
-    Expects a POST request with:
-    - user_id: ID of the user.
-    - manager_id: ID of the manager overseeing the user.
-    - interaction_type: The type of interaction (e.g., "disease_detection", "inventory_management").
-    - metadata: Optional. Additional details about the interaction.
-
-    Returns: JSON response indicating success or failure.
-    """
     try:
         user_id = request.json.get('user_id')
         manager_id = request.json.get('manager_id')
         interaction_type = request.json.get('interaction_type')
         metadata = request.json.get('metadata')
 
-        if not user_id or not manager_id or not interaction_type:
+        if not all([user_id, manager_id, interaction_type]):
             return jsonify({"error": "Missing required fields"}), 400
 
         return log_assistant_usage(user_id, manager_id, interaction_type, metadata)
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500

routes/health_routes.py

@@ -1,23 +1,20 @@
 # routes/health_routes.py
 
-from flask import Blueprint, request, jsonify
+from flask import Blueprint, jsonify
 from controllers.health_management import detect_and_log_disease, get_health_record, get_health_records_for_user
 
 # Create a Blueprint for the health-related routes
 health_bp = Blueprint('health', __name__)
 
-
 # Route to detect disease and log the health record
 @health_bp.route('/detect-disease', methods=['POST'])
 def detect_disease_route():
     """
     Detect a poultry disease and log the health record.
-
     Expects a POST request with the following fields:
     - user_id: ID of the user.
     - poultry_id: ID of the poultry.
     - image: Image of poultry feces for disease detection.
-
     Returns: JSON response with disease detection result and health record log status.
     """
     try:
@@ -31,9 +28,6 @@ def detect_disease_route():
 def get_health_record_route(record_id):
     """
     Retrieve a health record by its ID.
-
-    Expects a GET request with the health record ID in the URL.
-
     Returns: JSON response with the health record details.
     """
     try:
@@ -47,9 +41,6 @@ def get_health_record_route(record_id):
 def get_health_records_for_user_route(user_id):
     """
     Retrieve all health records for a specific user.
-
-    Expects a GET request with the user ID in the URL.
-
     Returns: JSON response with the list of health records for the user.
     """
     try:

routes/inventory_routes.py

@@ -1,20 +1,22 @@
 # routes/inventory_routes.py
 
 
-from flask import Blueprint, request, jsonify
-from controllers.inventory import add_inventory_item, update_inventory_item_by_id, get_inventory_by_id, \
+from flask import Blueprint, jsonify
+from controllers.inventory import (
+    add_inventory_item,
+    update_inventory_item_by_id,
+    get_inventory_by_id,
     get_all_inventory
+)
 
 # Create a Blueprint for the inventory-related routes
 inventory_bp = Blueprint('inventory', __name__)
 
-
 # Route to add a new inventory item
 @inventory_bp.route('/add-inventory-item', methods=['POST'])
 def add_inventory_item_route():
     """
     Add a new inventory item to the system.
-
     Expects a POST request with the following fields:
     - user_id: The ID of the user adding the item.
     - item_name: The name of the inventory item.
@@ -22,7 +24,6 @@ def add_inventory_item_route():
     - quantity: The total quantity of the item.
     - unit: The unit of measurement (e.g., "kg", "liters").
     - description: Optional. Additional description or notes about the item.
-
     Returns: JSON response with the status of the addition.
     """
     try:
@@ -36,12 +37,10 @@ def add_inventory_item_route():
 def update_inventory_item_route():
     """
     Update an existing inventory item in the system.
-
     Expects a PUT request with the following fields:
     - user_id: The ID of the user updating the item.
     - item_id: The ID of the inventory item to be updated.
     - updated_fields: Dictionary containing fields to be updated (e.g., quantity, used_quantity).
-
     Returns: JSON response with the status of the update.
     """
     try:
@@ -55,9 +54,7 @@ def update_inventory_item_route():
 def get_inventory_item_route(item_id):
     """
     Retrieve an inventory item by its ID.
-
     Expects a GET request with the item_id as a URL parameter.
-
     Returns: JSON response with the inventory item details.
     """
     try:
@@ -71,7 +68,6 @@ def get_inventory_item_route(item_id):
 def get_all_inventory_route():
     """
     Retrieve all inventory items in the system.
-
     Returns: JSON response with the list of all inventory items.
     """
     try:

routes/usage_routes.py

@@ -1,6 +1,6 @@
 # routes/usage_routes.py
 
-from flask import Blueprint, request, jsonify
+from flask import Blueprint, jsonify
 from controllers.usage_tracking import log_usage, get_usage_for_user, get_usage_for_manager, get_usage_by_type
 
 # Create a Blueprint for the usage-related routes
@@ -13,11 +13,11 @@ def log_usage_route():
     """
     Logs the usage of the assistant by a user.
 
-    Expects a POST request with the following fields:
+    Expects a POST request with:
     - user_id: The ID of the user interacting with the assistant.
     - manager_id: The ID of the user's manager.
     - interaction_type: The type of interaction (e.g., "disease_detection", "inventory_management").
-    - metadata: Optional. Additional details about the interaction (e.g., query details or results).
+    - metadata: Optional. Additional details about the interaction.
 
     Returns: JSON response indicating success or failure.
     """

services/disease_detection.py

@@ -4,34 +4,67 @@ import cv2
 import numpy as np
 from keras.models import load_model
 
-# Load the model (ensure the path is correct)
+# Load the model (ensure the model path is correct)
 my_model = load_model('models/Final_Chicken_disease_model.h5')
 
-# Disease names and recommendations
+# Mapping of disease labels to disease names, health status, and treatment recommendations
 name_disease = {0: 'Coccidiosis', 1: 'Healthy', 2: 'New Castle Disease', 3: 'Salmonella'}
 result = {0: 'Critical', 1: 'No issue', 2: 'Critical', 3: 'Critical'}
-recommend = {0: 'Panadol', 1: 'You have no need Medicine', 2: 'Percetamol', 3: 'Ponston'}
+recommend = {
+    0: 'Panadol',  # Treatment for Coccidiosis
+    1: 'You have no need for Medicine',  # Healthy case
+    2: 'Percetamol',  # Treatment for New Castle Disease
+    3: 'Ponston'  # Treatment for Salmonella
+}
 
 
 class PoultryFarmBot:
     def __init__(self, db):
-        self.db = db  # MongoDB database for future use
+        """
+        Initializes the PoultryFarmBot with a MongoDB connection.
+        :param db: MongoDB connection for future use in logging or storing results.
+        """
+        self.db = db
 
     def preprocess_image(self, image):
+        """
+        Preprocesses the input image for model prediction.
+        Resizes the image to 224x224 and expands the dimensions to match the model input format.
+
+        :param image: Input image (numpy array).
+        :return: Preprocessed image ready for prediction or None if an error occurs.
+        """
         try:
-            image_check = cv2.resize(image, (224, 224))
-            image_check = np.expand_dims(image_check, axis=0)
+            image_check = cv2.resize(image, (224, 224))  # Resize image
+            image_check = np.expand_dims(image_check, axis=0)  # Add batch dimension
             return image_check
         except Exception as e:
             print(f"Error in image preprocessing: {e}")
             return None
 
     def predict(self, image):
+        """
+        Predicts the disease from the input image using the pre-trained model.
+
+        :param image: Input image (numpy array).
+        :return: Tuple of predicted disease name, health status, and treatment recommendation.
+        """
+        # Preprocess the image
         image_check = self.preprocess_image(image)
+
         if image_check is None:
             return "Image preprocessing failed.", None, None, None
-        indx = my_model.predict(image_check).argmax()
-        name = name_disease.get(indx, "Unknown disease")
-        status = result.get(indx, "unknown condition")
-        recom = recommend.get(indx, "no recommendation available")
-        return name, status, recom
+
+        try:
+            # Predict disease using the loaded model
+            indx = my_model.predict(image_check).argmax()
+
+            # Get the disease name, status, and recommended treatment
+            name = name_disease.get(indx, "Unknown disease")
+            status = result.get(indx, "unknown condition")
+            recom = recommend.get(indx, "no recommendation available")
+
+            return name, status, recom
+        except Exception as e:
+            print(f"Error during disease prediction: {e}")
+            return "Prediction failed.", None, None, None

services/inventory_manager.py

@@ -1,7 +1,6 @@
 # services/inventory_manager.py
 
 
-from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
 
@@ -12,11 +11,9 @@ from config.db import db
 inventory_collection = db.inventory  # MongoDB collection to store inventory items
 
 
-# Inventory Management Service
-
 class InventoryManager:
     """
-    Inventory Manager Service to handle adding, updating, and retrieving inventory items.
+    Inventory Manager Service to handle adding, updating, retrieving, and deleting inventory items.
     """
 
     def __init__(self):
@@ -32,7 +29,7 @@ class InventoryManager:
         :param quantity: The total quantity of the item.
         :param unit: The unit of measurement (e.g., "kg", "liters").
         :param description: Optional. Additional description or notes about the item.
-        :return: The inserted inventory item's ID.
+        :return: The inserted inventory item's ID or None if there's an error.
         """
         try:
             new_item = {
@@ -60,7 +57,7 @@ class InventoryManager:
         :return: Boolean indicating whether the update was successful.
         """
         try:
-            updated_fields['updated_at'] = datetime.datetime.utcnow()  # Update timestamp
+            updated_fields['updated_at'] = datetime.datetime.utcnow()  # Update the timestamp
             result = self.collection.update_one(
                 {"_id": ObjectId(item_id)},
                 {"$set": updated_fields}
@@ -87,14 +84,13 @@ class InventoryManager:
         """
         Retrieve all inventory items, optionally filtered by user ID.
 
-        :param user_id: Optional. The ID of the user to filter by.
+        :param user_id: Optional. The ID of the user to filter the results by.
         :return: A list of inventory items or an empty list if none are found.
         """
         try:
             if user_id:
                 return list(self.collection.find({"user_id": ObjectId(user_id)}))
-            else:
-                return list(self.collection.find())
+            return list(self.collection.find())
         except Exception as e:
             print(f"Error retrieving inventory items: {e}")
             return []

services/llama_service.py

@@ -3,28 +3,39 @@
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load Llama 3.2 model and tokenizer for text generation
+# Load Llama model and tokenizer for text generation
 model_name = "meta-llama/Llama-3.2-1B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Set the padding token to EOS token or add a new padding token
+# Ensure the tokenizer has a padding token; if not, add a new padding token
 if tokenizer.pad_token is None:
     tokenizer.add_special_tokens({'pad_token': '[PAD]'})
     model.resize_token_embeddings(len(tokenizer))
 
 
 def llama2_response(user_input):
+    """
+    Generates a text response using the Llama model based on user input.
+
+    :param user_input: The input string from the user.
+    :return: A generated response string or an error message if something goes wrong.
+    """
     try:
+        # Tokenize the input text, ensuring truncation and padding
         inputs = tokenizer(user_input, return_tensors="pt", truncation=True, max_length=150, padding=True)
+
+        # Generate the model's response
         outputs = model.generate(
             inputs["input_ids"],
-            max_length=150,
-            do_sample=True,
-            temperature=0.7,
-            pad_token_id=tokenizer.pad_token_id,
-            attention_mask=inputs["attention_mask"]
+            max_length=150,  # Set a reasonable max length for the response
+            do_sample=True,  # Enable sampling for varied responses
+            temperature=0.7,  # Control randomness of sampling
+            pad_token_id=tokenizer.pad_token_id,  # Ensure the use of a valid pad token
+            attention_mask=inputs["attention_mask"]  # Use attention mask for padded tokens
         )
+
+        # Decode the generated response, skipping special tokens
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         return response
     except Exception as e:

services/report_generator.py

@@ -1,17 +1,14 @@
 # services/report_generator.py
 
 
-from pymongo import MongoClient
-from bson.objectid import ObjectId
 import pandas as pd
+from bson.objectid import ObjectId
 import datetime
-
-# MongoDB Setup (import from your db configuration)
 from config.db import db
 
 # Define collections for report generation
 inventory_collection = db.inventory
-health_collection = db.health_records
+health_collection = db.health
 usage_collection = db.user_usage
 
 
@@ -39,13 +36,9 @@ class ReportGenerator:
             if not inventory_items:
                 return pd.DataFrame([])
 
-            # Convert the MongoDB documents to a pandas DataFrame
+            # Convert MongoDB documents to a pandas DataFrame
             df = pd.DataFrame(inventory_items)
             df = df.drop(columns=['_id', 'user_id'])  # Drop internal fields
-
-            # Add a column for total value (if relevant for future use)
-            # df['total_value'] = df['quantity'] * df['unit_price']
-
             return df
         except Exception as e:
             print(f"Error generating inventory report: {e}")
@@ -65,10 +58,9 @@ class ReportGenerator:
             if not health_records:
                 return pd.DataFrame([])
 
-            # Convert the MongoDB documents to a pandas DataFrame
+            # Convert MongoDB documents to a pandas DataFrame
             df = pd.DataFrame(health_records)
-            df = df.drop(columns=['_id', 'user_id'])
-
+            df = df.drop(columns=['_id', 'user_id'])  # Drop internal fields
             return df
         except Exception as e:
             print(f"Error generating health report: {e}")
@@ -97,10 +89,9 @@ class ReportGenerator:
             if not usage_records:
                 return pd.DataFrame([])
 
-            # Convert the MongoDB documents to a pandas DataFrame
+            # Convert MongoDB documents to a pandas DataFrame
             df = pd.DataFrame(usage_records)
-            df = df.drop(columns=['_id', 'user_id', 'manager_id'])
-
+            df = df.drop(columns=['_id', 'user_id', 'manager_id'])  # Drop internal fields
             return df
         except Exception as e:
             print(f"Error generating usage report: {e}")

services/usage_tracker.py

@@ -3,8 +3,6 @@
 from pymongo import MongoClient
 from bson.objectid import ObjectId
 import datetime
-
-# MongoDB Setup (import from your db configuration)
 from config.db import db
 
 # Define the usage tracking collection

templates/dashboard.html

@@ -0,0 +1 @@
+<!--template/dashboard.html-->

templates/index.html

@@ -1,3 +1,4 @@
+<!--template/index.html-->
 <!DOCTYPE html>
 <html dir="ltr" lang="en">
 <head>

templates/login.html

@@ -0,0 +1 @@
+<!--template/login.html-->

templates/register.html

@@ -0,0 +1 @@
+<!--template/register.html-->

templates/usage_report.html

@@ -0,0 +1 @@
+<!--template/usage_report.html-->