update space

.idea/workspace.xml

@@ -5,8 +5,10 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="27c9ae1a-a6fa-4472-8bcd-a7087620894b" name="Changes" comment="update space">
+      <change afterPath="$PROJECT_DIR$/models/schemas/inquiry_schema.py" afterDir="false" />
+      <change afterPath="$PROJECT_DIR$/routes/chatbot.py" afterDir="false" />
+      <change afterPath="$PROJECT_DIR$/templates/chatbot.html" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/templates/main/base.html" beforeDir="false" afterPath="$PROJECT_DIR$/templates/main/base.html" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -19,8 +21,8 @@
         <option value="Dockerfile" />
         <option value="JavaScript File" />
         <option value="CSS File" />
-        <option value="Python Script" />
         <option value="HTML File" />
+        <option value="Python Script" />
       </list>
     </option>
   </component>
@@ -125,15 +127,7 @@
       <workItem from="1730378091154" duration="2435000" />
       <workItem from="1730381153348" duration="8786000" />
       <workItem from="1730397485849" duration="22781000" />
-      <workItem from="1730454506390" duration="6619000" />
-    </task>
-    <task id="LOCAL-00053" summary="update space">
-      <option name="closed" value="true" />
-      <created>1730378908005</created>
-      <option name="number" value="00053" />
-      <option name="presentableId" value="LOCAL-00053" />
-      <option name="project" value="LOCAL" />
-      <updated>1730378908005</updated>
+      <workItem from="1730454506390" duration="8819000" />
     </task>
     <task id="LOCAL-00054" summary="update space">
       <option name="closed" value="true" />
@@ -519,7 +513,15 @@
       <option name="project" value="LOCAL" />
       <updated>1730465119418</updated>
     </task>
-    <option name="localTasksCounter" value="102" />
+    <task id="LOCAL-00102" summary="update space">
+      <option name="closed" value="true" />
+      <created>1730468175296</created>
+      <option name="number" value="00102" />
+      <option name="presentableId" value="LOCAL-00102" />
+      <option name="project" value="LOCAL" />
+      <updated>1730468175296</updated>
+    </task>
+    <option name="localTasksCounter" value="103" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

models/schemas/inquiry_schema.py

@@ -0,0 +1,41 @@
+# models/schemas/inquiry_schema.py
+
+from pydantic import BaseModel, Field
+from datetime import datetime
+from typing import Optional, List
+from bson import ObjectId
+
+class PyObjectId(ObjectId):
+    @classmethod
+    def __get_validators__(cls):
+        yield cls.validate
+
+    @classmethod
+    def validate(cls, v):
+        if not ObjectId.is_valid(v):
+            raise ValueError("Invalid ObjectId")
+        return ObjectId(v)
+
+class Inquiry(BaseModel):
+    id: PyObjectId = Field(default_factory=PyObjectId, alias="_id")
+    user_id: str
+    inquiry_text: str
+    response_text: str
+    timestamp: datetime = Field(default_factory=datetime.utcnow)
+    attached_image: Optional[str] = None  # File path or image ID in storage
+    detected_disease: Optional[str] = None
+    disease_confidence: Optional[float] = None
+
+    class Config:
+        json_encoders = {ObjectId: str}
+        schema_extra = {
+            "example": {
+                "user_id": "user123",
+                "inquiry_text": "What is the treatment for coccidiosis?",
+                "response_text": "The recommended treatment for coccidiosis is...",
+                "timestamp": "2024-11-01T12:00:00Z",
+                "attached_image": "image123.jpg",
+                "detected_disease": "Coccidiosis",
+                "disease_confidence": 85.5
+            }
+        }

routes/chatbot.py

@@ -0,0 +1,117 @@
+# routes/chatbot.py
+
+import io
+
+import cv2
+import numpy as np
+from fastapi.templating import Jinja2Templates
+from services.utils import db
+from fastapi import APIRouter, WebSocket, WebSocketDisconnect, UploadFile, File, Depends, Request
+from fastapi.responses import HTMLResponse, JSONResponse
+from datetime import datetime
+from services.disease_detection_service import bot
+from models.schemas.inquiry_schema import Inquiry
+from services.auth_service import get_current_user
+from typing import List
+
+chatbot_router = APIRouter()
+
+templates = Jinja2Templates(directory="templates")
+
+# MongoDB collection for inquiries
+inquiry_collection = db.get_collection("inquiries")
+
+# Store active WebSocket connections
+active_connections: List[WebSocket] = []
+
+async def greeting_based_on_time():
+    current_hour = datetime.now().hour
+    if 5 <= current_hour < 12:
+        return "Good morning"
+    elif 12 <= current_hour < 17:
+        return "Good afternoon"
+    else:
+        return "Good evening"
+
+@chatbot_router.get("/", response_class=HTMLResponse)
+async def chatbot_page(request: Request, current_user: str = Depends(get_current_user)):
+    greeting = await greeting_based_on_time()
+    context = {"request": request, "greeting": f"{greeting}, {current_user}! How can I help you today?"}
+    return templates.TemplateResponse("chatbot.html", context)
+
+@chatbot_router.websocket("/ws/chat")
+async def websocket_endpoint(websocket: WebSocket, current_user: str = Depends(get_current_user)):
+    """Handle real-time WebSocket connections for chat and sidebar updates."""
+    await websocket.accept()
+    active_connections.append(websocket)
+    try:
+        while True:
+            data = await websocket.receive_text()
+            response_text, detected_disease, confidence = await handle_inquiry(data, current_user)
+            await websocket.send_json({"response": response_text, "disease": detected_disease, "confidence": confidence})
+            await broadcast_inquiry_history(current_user)  # Update sidebar for all users
+    except WebSocketDisconnect:
+        active_connections.remove(websocket)
+
+async def handle_inquiry(text: str, current_user: str, file: UploadFile = None):
+    """Process inquiries asynchronously and update MongoDB."""
+    detected_disease, disease_confidence = None, None
+    response_text = ""
+
+    # Handle image inquiries
+    if file:
+        image_data = await file.read()
+        image_np = np.frombuffer(image_data, np.uint8)
+        response_text, detected_disease, status, recommendation, disease_confidence = bot.diagnose_and_respond(
+            cv2.imdecode(image_np, cv2.IMREAD_COLOR)
+        )
+    else:
+        response_text = bot.llama_response(text)
+
+    # Save inquiry to MongoDB
+    inquiry = Inquiry(
+        user_id=current_user,
+        inquiry_text=text,
+        response_text=response_text,
+        attached_image=file.filename if file else None,
+        detected_disease=detected_disease,
+        disease_confidence=disease_confidence
+    )
+    inquiry_collection.insert_one(inquiry.dict(by_alias=True))
+    return response_text, detected_disease, disease_confidence
+
+async def broadcast_inquiry_history(user_id: str):
+    """Send updated inquiry history to all WebSocket clients."""
+    history = list(inquiry_collection.find({"user_id": user_id}).sort("timestamp", -1).limit(10))
+    for connection in active_connections:
+        await connection.send_json({"history": history})
+
+@chatbot_router.post("/inquire")
+async def inquire(text: str, current_user: str = Depends(get_current_user), file: UploadFile = None):
+    """Handle inquiries with text and optional image for disease detection."""
+    response_text = ""
+    detected_disease, disease_confidence = None, None
+
+    # Handle image inquiries with disease detection
+    if file:
+        image_data = await file.read()
+        image_np = np.frombuffer(image_data, np.uint8)  # Convert to NumPy array
+        response, disease_name, status, recommendation, confidence = bot.diagnose_and_respond(cv2.imdecode(image_np, cv2.IMREAD_COLOR))
+        response_text = response
+        detected_disease, disease_confidence = disease_name, confidence
+    else:
+        # General inquiries using Llama model
+        response_text = bot.llama_response(text)
+
+    # Log inquiry into MongoDB
+    inquiry = Inquiry(
+        user_id=current_user,
+        inquiry_text=text,
+        response_text=response_text,
+        attached_image=file.filename if file else None,
+        detected_disease=detected_disease,
+        disease_confidence=disease_confidence
+    )
+    inquiry_collection.insert_one(inquiry.dict(by_alias=True))
+
+    return JSONResponse(content={"response_text": response_text, "detected_disease": detected_disease, "confidence": disease_confidence})

templates/chatbot.html

@@ -0,0 +1,93 @@
+<!-- templates/chatbot.html -->
+
+{% extends "base.html" %}
+{% block title %}Poultry Management Chatbot{% endblock %}
+
+{% block content %}
+<div class="container mt-5">
+    <div class="row">
+        <!-- Sidebar for Inquiry History -->
+        <div class="col-md-3">
+            <div class="card">
+                <div class="card-header">
+                    <h5>Inquiry History</h5>
+                </div>
+                <div id="inquiry-history" class="list-group" style="height: 60vh; overflow-y: auto;">
+                    <!-- History items will be dynamically added here -->
+                </div>
+            </div>
+        </div>
+
+        <!-- Main Chat Area -->
+        <div class="col-md-9">
+            <div id="chat-window" class="card p-4" style="height: 60vh; overflow-y: auto;">
+                <div class="text-center mb-3">
+                    <h5>{{ greeting }}</h5>
+                </div>
+                <div id="chat-content"></div>
+            </div>
+            <!-- Chat Input -->
+            <form id="chat-form" class="mt-3" enctype="multipart/form-data">
+                <div class="input-group">
+                    <input type="text" class="form-control" id="user-input" placeholder="Type your message..." required>
+                    <input type="file" id="file-upload" accept="image/*" style="display: none;">
+                    <button class="btn btn-outline-secondary" type="button" onclick="document.getElementById('file-upload').click()">📎</button>
+                    <button class="btn btn-primary" type="submit">Send</button>
+                </div>
+            </form>
+        </div>
+    </div>
+</div>
+
+<script>
+    const chatSocket = new WebSocket(`ws://${window.location.host}/ws/chat`);
+
+    chatSocket.onmessage = function(event) {
+        const data = JSON.parse(event.data);
+
+        // Display response
+        if (data.response) {
+            displayMessage(data.response, "assistant");
+            if (data.disease) {
+                displayMessage(`Detected disease: ${data.disease} with confidence: ${data.confidence}%`, "assistant");
+            }
+        }
+
+        // Update sidebar with inquiry history
+        if (data.history) {
+            updateInquiryHistory(data.history);
+        }
+    };
+
+    document.getElementById("chat-form").addEventListener("submit", async function(e) {
+        e.preventDefault();
+        const userMessage = document.getElementById("user-input").value;
+        displayMessage(userMessage, "user");
+
+        // Send message to WebSocket
+        chatSocket.send(userMessage);
+
+        document.getElementById("user-input").value = "";  // Clear input
+    });
+
+    function displayMessage(message, sender) {
+        const messageBubble = document.createElement("div");
+        messageBubble.classList.add("p-3", "rounded", sender === "user" ? "bg-primary text-white text-right" : "bg-light text-dark");
+        messageBubble.style.maxWidth = "70%";
+        messageBubble.style.margin = sender === "user" ? "10px auto 10px 50px" : "10px 50px 10px auto";
+        messageBubble.textContent = message;
+        document.getElementById("chat-content").appendChild(messageBubble);
+    }
+
+    function updateInquiryHistory(history) {
+        const historyDiv = document.getElementById("inquiry-history");
+        historyDiv.innerHTML = "";  // Clear existing history
+        history.forEach(inquiry => {
+            const historyItem = document.createElement("div");
+            historyItem.classList.add("list-group-item");
+            historyItem.textContent = inquiry.inquiry_text;
+            historyDiv.appendChild(historyItem);
+        });
+    }
+</script>
+{% endblock %}