update space

.idea/misc.xml

@@ -4,4 +4,7 @@
     <option name="sdkName" value="Python 3.11 (Poultry_Diseases_Detector)" />
   </component>
   <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12 (Generative_AI_with_poultry_disease_detection_system)" project-jdk-type="Python SDK" />
+  <component name="PythonCompatibilityInspectionAdvertiser">
+    <option name="version" value="3" />
+  </component>
 </project>

app.py

@@ -7,14 +7,14 @@ from pymongo import MongoClient, errors
 from datetime import datetime
 from werkzeug.security import generate_password_hash
 from services.utils import PoultryFarmBot
-from services.chatbot import build_chatbot_interface
+from services.chatbot_interface import build_chatbot_interface
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from flask import Flask, render_template, request, redirect, url_for, session
 from celery import Celery
 import time
 import threading
 import gunicorn.app.base
-from gunicorn.six import iteritems
+from six import iteritems
 
 # Setup logging for better monitoring
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -48,12 +48,14 @@ for attempt in range(max_retries):
         enquiries_collection = db.enquiries  # Collection to store farmer enquiries
         users_collection = db.users  # Collection to store user credentials
         logs_collection = db.logs  # Collection to store application logs
+        logger.info("Successfully connected to MongoDB.")
         break
     except errors.ServerSelectionTimeoutError as e:
         logger.error(f"Failed to connect to MongoDB (Attempt {attempt + 1}/{max_retries}): {e}")
         if attempt < max_retries - 1:
             time.sleep(5)  # Wait for 5 seconds before retrying
         else:
+            logger.critical("Could not connect to MongoDB after multiple attempts. Exiting application.")
             raise ConnectionError("Could not connect to MongoDB. Please check the MONGO_URI and ensure the database is running.")
 
 def log_to_db(level, message):
@@ -64,6 +66,7 @@ def log_to_db(level, message):
             "timestamp": datetime.utcnow()
         }
         logs_collection.insert_one(log_entry)
+        logger.debug(f"Logged to database: {log_entry}")
     except Exception as e:
         logger.error(f"Failed to log to database: {e}")
 
@@ -93,6 +96,7 @@ else:
 # Initialize the bot instance
 logger.info("Initializing PoultryFarmBot instance.")
 bot = PoultryFarmBot(db)
+logger.info("PoultryFarmBot instance initialized.")
 
 # Global model and tokenizer variables
 model = None
@@ -131,10 +135,13 @@ def authenticate_user(username, password):
     Returns:
         bool: True if authentication is successful, False otherwise.
     """
+    logger.info(f"Authenticating user: {username}")
     user = bot.authenticate_user(username, password)
     if user:
+        logger.info(f"Authentication successful for user: {username}")
         return True, user
     else:
+        logger.warning(f"Authentication failed for user: {username}")
         return False, None
 
 # Registration function
@@ -149,6 +156,7 @@ def register_user(username, password):
     Returns:
         bool: True if registration is successful, False otherwise.
     """
+    logger.info(f"Registering user: {username}")
     try:
         if users_collection.find_one({"username": username}):
             logger.warning("Username already exists: %s", username)
@@ -167,12 +175,14 @@ def login():
     if request.method == 'POST':
         username = request.form['username']
         password = request.form['password']
+        logger.info(f"Login attempt for user: {username}")
         success, user = authenticate_user(username, password)
         if success:
             logger.info("Authentication successful for user: %s", username)
             session['username'] = username
 
             # Start loading the model asynchronously after successful login
+            logger.info("Starting model loading asynchronously.")
             load_model_and_tokenizer.apply_async()
 
             return redirect(url_for('chatbot'))
@@ -186,6 +196,7 @@ def register():
     if request.method == 'POST':
         username = request.form['username']
         password = request.form['password']
+        logger.info(f"Registration attempt for user: {username}")
         success = register_user(username, password)
         if success:
             logger.info("Registration successful for user: %s", username)
@@ -198,14 +209,18 @@ def register():
 @app.route('/chatbot')
 def chatbot():
     if 'username' not in session:
+        logger.warning("Access to chatbot attempted without login. Redirecting to login page.")
         return redirect(url_for('login'))
     username = session['username']
 
     # Wait until the model is loaded with a timeout mechanism
+    logger.info("Waiting for model to load.")
     if not model_loading_event.wait(timeout=100):  # Timeout after 100 seconds
+        logger.error("Model loading timed out for user: %s", username)
         return "Model loading timed out. Please try again later."
 
     if model is None or tokenizer is None:
+        logger.error("Model failed to load for user: %s", username)
         return "Model failed to load. Please try again later."
 
     logger.info("Launching Gradio chatbot interface for user: %s", username)
@@ -230,6 +245,7 @@ class StandaloneApplication(gunicorn.app.base.BaseApplication):
 # Launch the Flask application using Gunicorn in a production-safe manner
 if __name__ == "__main__":
     try:
+        logger.info("Launching Flask server for AdminLTE authentication.")
         options = {
             'bind': '%s:%s' % ('0.0.0.0', '5000'),
             'workers': 2,  # Set number of worker processes
@@ -239,4 +255,4 @@ if __name__ == "__main__":
         StandaloneApplication(app, options).run()
     except Exception as e:
         logger.error(f"Failed to launch Flask server: {e}")
-        raise RuntimeError("Could not launch the Flask server. Please check the application setup.")
+        raise RuntimeError("Could not launch the Flask server. Please check the application setup.")

requirements.txt

@@ -4,6 +4,7 @@ keras==2.9.0
 torch
 torchvision
 numpy==1.24.3
+
 # Transformers and AI model handling
 transformers
 accelerate
@@ -11,16 +12,14 @@ accelerate
 # Image Processing Libraries
 opencv-python
 
-
 # Data Handling and Analysis
 pandas
 celery
 gunicorn
+
 # OpenAI Integration
 openai
-Werkzeug
+werkzeug
 gradio
 flask
 pymongo
-
-