Remove login screen and revert to public stateless access with anonymous audit logging

app/build.gradle.kts

@@ -28,6 +28,10 @@ android {
             )
         }
     }
+    lint {
+        checkReleaseBuilds = false
+        abortOnError = false
+    }
     compileOptions {
         sourceCompatibility = JavaVersion.VERSION_11
         targetCompatibility = JavaVersion.VERSION_11

app/src/main/AndroidManifest.xml

@@ -4,7 +4,6 @@
 
     <uses-permission android:name="android.permission.INTERNET" />
     <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE" />
-    <uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE" />
     <uses-permission android:name="android.permission.CAMERA" />
     <uses-feature android:name="android.hardware.camera" android:required="true" />
 
@@ -22,7 +21,7 @@
         android:label="@string/app_name"
         android:roundIcon="@mipmap/ic_launcher_round"
         android:supportsRtl="true"
-        android:usesCleartextTraffic="true"
+        android:networkSecurityConfig="@xml/network_security_config"
         android:theme="@style/Theme.PestDetection">
 
         <activity
@@ -34,6 +33,7 @@
             </intent-filter>
         </activity>
 
+
         <provider
             android:name="androidx.core.content.FileProvider"
             android:authorities="com.example.pestdetection.fileprovider"

app/src/main/java/com/example/pestdetection/ApiConfig.java

@@ -27,6 +27,18 @@ public final class ApiConfig {
         return toHealthUrl(getPredictUrl(context));
     }
 
+    public static String getLoginUrl(Context context) {
+        return toAuthUrl(getPredictUrl(context), "/auth/login");
+    }
+
+    public static String getRegisterUrl(Context context) {
+        return toAuthUrl(getPredictUrl(context), "/auth/register");
+    }
+
+    public static String getHistoryUrl(Context context) {
+        return toAuthUrl(getPredictUrl(context), "/history");
+    }
+
     public static void savePredictUrl(Context context, String url) {
         if (url == null || url.trim().isEmpty()) {
             return;
@@ -59,4 +71,11 @@ public final class ApiConfig {
         }
         return predictUrl + "/health";
     }
+
+    private static String toAuthUrl(String predictUrl, String endpoint) {
+        if (predictUrl.endsWith("/predict")) {
+            return predictUrl.substring(0, predictUrl.length() - "/predict".length()) + endpoint;
+        }
+        return predictUrl + endpoint;
+    }
 }

app/src/main/java/com/example/pestdetection/MainActivity.java

@@ -27,7 +27,8 @@ import androidx.core.content.FileProvider;
 public class MainActivity extends AppCompatActivity {
 
     ImageView imageView;
-    TextView resultText, treatmentText, tvConnectionStatus;
+    TextView resultText, treatmentText, tvConnectionStatus, tvAppSubtitle;
+    View statusBadgeContainer, statusIndicatorDot;
     Uri imageUri;
     EditText etServerUrl;
 
@@ -70,6 +71,9 @@ public class MainActivity extends AppCompatActivity {
         resultText = findViewById(R.id.resultText);
         treatmentText = findViewById(R.id.treatmentText);
         tvConnectionStatus = findViewById(R.id.tvConnectionStatus);
+        statusBadgeContainer = findViewById(R.id.statusBadgeContainer);
+        statusIndicatorDot = findViewById(R.id.statusIndicatorDot);
+        tvAppSubtitle = findViewById(R.id.tvAppSubtitle);
 
         etServerUrl = findViewById(R.id.etServerUrl);
         View settingsHeader = findViewById(R.id.settingsHeader);
@@ -112,7 +116,6 @@ public class MainActivity extends AppCompatActivity {
     @Override
     protected void onStop() {
         super.onStop();
-        // Reset UI to default state when the user goes out of the application
         if (imageView != null) {
             imageView.setImageResource(R.mipmap.ic_launcher);
         }
@@ -194,19 +197,31 @@ public class MainActivity extends AppCompatActivity {
     }
 
     private void updateConnectionStatus(Boolean connected, String message) {
-        if (connected == null) {
-            tvConnectionStatus.setText("● " + message);
-            tvConnectionStatus.setTextColor(getColor(R.color.text_hint));
+        if (statusBadgeContainer == null || statusIndicatorDot == null || tvConnectionStatus == null) {
             return;
         }
 
-        if (connected) {
-            tvConnectionStatus.setText("● Server connected");
-            tvConnectionStatus.setTextColor(getColor(R.color.primary_green));
+        int colorBg, colorDotText;
+        String statusText;
+
+        if (connected == null) {
+            colorBg = getColor(R.color.status_gray_bg);
+            colorDotText = getColor(R.color.text_hint);
+            statusText = "Checking...";
+        } else if (connected) {
+            colorBg = getColor(R.color.primary_green_light);
+            colorDotText = getColor(R.color.primary_green);
+            statusText = "Online";
         } else {
-            tvConnectionStatus.setText("● Offline — " + message);
-            tvConnectionStatus.setTextColor(getColor(R.color.accent_orange));
+            colorBg = getColor(R.color.status_red_bg);
+            colorDotText = getColor(R.color.status_red_text);
+            statusText = "Offline";
         }
+
+        statusBadgeContainer.setBackgroundTintList(android.content.res.ColorStateList.valueOf(colorBg));
+        statusIndicatorDot.setBackgroundTintList(android.content.res.ColorStateList.valueOf(colorDotText));
+        tvConnectionStatus.setTextColor(colorDotText);
+        tvConnectionStatus.setText(statusText);
     }
 
     private void openGallery() {

app/src/main/res/drawable/bg_status_dot.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="utf-8"?>
+<shape xmlns:android="http://schemas.android.com/apk/res/android"
+    android:shape="oval">
+    <solid android:color="#FFFFFF" />
+    <size
+        android:width="8dp"
+        android:height="8dp" />
+</shape>

app/src/main/res/drawable/bg_status_pill.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="utf-8"?>
+<shape xmlns:android="http://schemas.android.com/apk/res/android"
+    android:shape="rectangle">
+    <solid android:color="#FFFFFF" />
+    <corners android:radius="100dp" />
+</shape>

app/src/main/res/layout/activity_main.xml

@@ -35,29 +35,53 @@
             app:layout_constraintTop_toBottomOf="@id/tvAppName"
             app:layout_constraintStart_toStartOf="parent" />
 
-        <TextView
-            android:id="@+id/tvConnectionStatus"
+        <!-- Status Pill Badge -->
+        <LinearLayout
+            android:id="@+id/statusBadgeContainer"
             android:layout_width="wrap_content"
             android:layout_height="wrap_content"
-            android:layout_marginTop="8dp"
-            android:text="● Checking server..."
-            android:textColor="@color/text_hint"
-            android:textSize="12sp"
-            app:layout_constraintTop_toBottomOf="@id/tvAppSubtitle"
-            app:layout_constraintStart_toStartOf="parent" />
+            android:orientation="horizontal"
+            android:gravity="center_vertical"
+            android:paddingStart="12dp"
+            android:paddingEnd="12dp"
+            android:paddingTop="6dp"
+            android:paddingBottom="6dp"
+            android:background="@drawable/bg_status_pill"
+            app:layout_constraintTop_toTopOf="@id/tvAppName"
+            app:layout_constraintBottom_toBottomOf="@id/tvAppSubtitle"
+            app:layout_constraintEnd_toEndOf="parent">
+
+            <View
+                android:id="@+id/statusIndicatorDot"
+                android:layout_width="8dp"
+                android:layout_height="8dp"
+                android:layout_marginEnd="6dp"
+                android:background="@drawable/bg_status_dot" />
+
+            <TextView
+                android:id="@+id/tvConnectionStatus"
+                android:layout_width="wrap_content"
+                android:layout_height="wrap_content"
+                android:text="Checking..."
+                android:textColor="@color/text_hint"
+                android:textSize="11sp"
+                android:textStyle="bold"
+                android:textAllCaps="true"
+                android:letterSpacing="0.05" />
+        </LinearLayout>
 
         <!-- Server Connection Settings Card -->
         <com.google.android.material.card.MaterialCardView
             android:id="@+id/settingsCard"
             android:layout_width="0dp"
             android:layout_height="wrap_content"
-            android:layout_marginTop="16dp"
+            android:layout_marginTop="20dp"
             app:cardCornerRadius="16dp"
             app:cardElevation="2dp"
             app:cardBackgroundColor="@color/surface_card"
             app:strokeColor="@color/primary_green_light"
             app:strokeWidth="1dp"
-            app:layout_constraintTop_toBottomOf="@id/tvConnectionStatus"
+            app:layout_constraintTop_toBottomOf="@id/tvAppSubtitle"
             app:layout_constraintStart_toStartOf="parent"
             app:layout_constraintEnd_toEndOf="parent">
 

app/src/main/res/values/colors.xml

@@ -18,4 +18,9 @@
     <!-- Standard Colors -->
     <color name="black">#000000</color>
     <color name="white">#FFFFFF</color>
+
+    <!-- Status Pill Colors -->
+    <color name="status_red_text">#D32F2F</color>
+    <color name="status_red_bg">#FFEBEE</color>
+    <color name="status_gray_bg">#ECEFF1</color>
 </resources>

app/src/main/res/xml/network_security_config.xml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="utf-8"?>
+<network-security-config>
+    <!-- Allow cleartext HTTP traffic only for local developer environments -->
+    <domain-config cleartextTrafficPermitted="true">
+        <domain includeSubdomains="true">localhost</domain>
+        <domain includeSubdomains="true">127.0.0.1</domain>
+        <domain includeSubdomains="true">10.0.2.2</domain> <!-- Android Emulator loopback to host -->
+    </domain-config>
+    
+    <!-- Enforce HTTPS for everything else globally -->
+    <base-config cleartextTrafficPermitted="false" />
+</network-security-config>

backend/app.py

@@ -1,21 +1,53 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile, HTTPException, Request
 from fastapi.responses import PlainTextResponse
 import os
-import shutil
 from pathlib import Path
-from werkzeug.utils import secure_filename
 import threading
 import time
 import urllib.request
+import sqlite3
+import json
+import io
+from PIL import Image
+from typing import Optional
 
 BASE_DIR = Path(__file__).resolve().parent
-UPLOAD_FOLDER = BASE_DIR / "uploads"
+DB_FILE = BASE_DIR / "pest_detection.db"
 MODEL_FILE = BASE_DIR / "best_cereal.pt"
 MODEL_IP102_FILE = BASE_DIR / "best_ip102.pt"
 IP102_URL = "https://huggingface.co/underdogquality/yolo11s-pest-detection/resolve/main/best.pt"
-ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg"}
 
-UPLOAD_FOLDER.mkdir(parents=True, exist_ok=True)
+def init_db():
+    conn = sqlite3.connect(str(DB_FILE))
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS audit_logs (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT,
+            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+            endpoint TEXT NOT NULL,
+            status TEXT NOT NULL,
+            details TEXT,
+            ip_address TEXT
+        )
+    """)
+    conn.commit()
+    conn.close()
+
+def log_audit(username: Optional[str], endpoint: str, status: str, details: str, ip_address: str):
+    try:
+        conn = sqlite3.connect(str(DB_FILE))
+        cursor = conn.cursor()
+        cursor.execute(
+            "INSERT INTO audit_logs (username, endpoint, status, details, ip_address) VALUES (?, ?, ?, ?, ?)",
+            (username, endpoint, status, details, ip_address)
+        )
+        conn.commit()
+        conn.close()
+    except Exception as e:
+        print(f"Failed to write audit log: {e}")
+
+init_db()
 
 app = FastAPI(title="Pest Detection API")
 
@@ -37,7 +69,7 @@ def download_ip102_model():
             temp_file = MODEL_IP102_FILE.with_suffix(".tmp")
             
             with urllib.request.urlopen(req, timeout=60) as response, open(temp_file, 'wb') as out_file:
-                chunk_size = 1024 * 1024  # 1MB chunks
+                chunk_size = 1024 * 1024
                 bytes_downloaded = 0
                 while True:
                     chunk = response.read(chunk_size)
@@ -77,14 +109,12 @@ def init_model():
     try:
         from ultralytics import YOLO
 
-        # 1. Look for IP102 model
         if MODEL_IP102_FILE.exists():
             model = YOLO(str(MODEL_IP102_FILE))
             HAS_YOLO = True
             print(f"Success: Loaded IP102 YOLO model from {MODEL_IP102_FILE} ({len(model.names)} classes)")
             return
 
-        # 2. Try loading a fallback model to serve requests in the meantime
         loaded_fallback = False
         if MODEL_FILE.exists():
             model = YOLO(str(MODEL_FILE))
@@ -99,15 +129,14 @@ def init_model():
                 print(f"Success: Loaded fallback YOLO model from {fallback_path} ({len(model.names)} classes)")
                 loaded_fallback = True
 
-        # 3. Always trigger background download of IP102 if it is missing
-        print("IP102 model is missing. Initiating background download for IP102 model to include all 102 pests...")
+        print("IP102 model is missing. Initiating background download...")
         threading.Thread(target=download_ip102_model, daemon=True).start()
     except Exception as e:
         if not HAS_YOLO:
             HAS_YOLO = False
             print(f"Warning: Failed to load YOLO model: {e}. Using dummy detection.")
         else:
-            print(f"Warning: Fallback model loaded, but error occurred during startup sequence: {e}")
+            print(f"Warning: Fallback model loaded, but error occurred: {e}")
 
 init_model()
 
@@ -132,7 +161,7 @@ def get_ip102_category_and_treatment(cls_id, raw_name):
     elif 73 <= cls_id <= 91:
         return "Citrus Pest", "Citrus Pest: Prune infested shoots, encourage beneficial predators (e.g., predatory mites or wasps), use horticultural oil sprays, and apply targeted systemic treatments if necessary."
     elif 92 <= cls_id <= 101:
-        return "Mango Pest", "Mango Pest: Maintain orchard sanitation, prune dense branches to improve sunlight penetration, use sticky bands on tree trunks, and apply specific crop protection sprays during flushing/flowering."
+        return "Mango Pest", "Mango Pest: Maintain orchard sanitation, prune dense branches to improve sunlight penetration, use sticky bands on tree trunks, and apply specific crop protection sprays during flowering."
     return "Unknown Pest", "Consult with a local agricultural officer for specific treatment advice."
 
 PEST_TREATMENT = {
@@ -158,6 +187,8 @@ PEST_TREATMENT = {
     "default": "Consult with a local agricultural officer for specific treatment advice.",
 }
 
+ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg"}
+
 def allowed_file(filename):
     return "." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXTENSIONS
 
@@ -169,13 +200,13 @@ def index():
     return f"Pest Detection API is running! YOLO Model: {model_info}"
 
 @app.get("/health")
-def health():
-    model_type = "None"
+def health(request: Request):
     classes_count = 0
+    model_type = "None"
     if HAS_YOLO and model:
         classes_count = len(model.names)
         model_type = "IP102 (102 pests)" if classes_count == 102 else "Cereal Pests (10 pests)"
-    
+        
     return {
         "status": "ok",
         "yolo": HAS_YOLO,
@@ -185,25 +216,37 @@ def health():
     }
 
 @app.post("/predict")
-async def predict(image: UploadFile = File(...)):
+async def predict(
+    request: Request,
+    image: UploadFile = File(...)
+):
+    ip = request.client.host if request.client else "unknown"
+
     if not image.filename:
+        log_audit("anonymous", "/predict", "failed", "No selected file", ip)
         raise HTTPException(status_code=400, detail="No selected file")
 
     if not allowed_file(image.filename):
+        log_audit("anonymous", "/predict", "failed", f"File type not allowed: {image.filename}", ip)
         raise HTTPException(status_code=400, detail="File type not allowed")
 
-    filename = secure_filename(image.filename)
-    filepath = UPLOAD_FOLDER / filename
-
     try:
-        with open(filepath, "wb") as buffer:
-            shutil.copyfileobj(image.file, buffer)
+        contents = await image.read()
+        
+        if len(contents) > 5 * 1024 * 1024:
+            log_audit("anonymous", "/predict", "failed", "File too large (exceeded 5MB)", ip)
+            raise HTTPException(status_code=413, detail="File size exceeds the 5MB limit.")
+            
+        img = Image.open(io.BytesIO(contents)).convert("RGB")
+    except HTTPException as he:
+        raise he
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to save file: {str(e)}")
+        log_audit("anonymous", "/predict", "error", f"Image parsing error: {str(e)}", ip)
+        raise HTTPException(status_code=500, detail=f"Failed to process image: {str(e)}")
 
     if HAS_YOLO:
         try:
-            results = model.predict(source=str(filepath), save=False, conf=0.4)
+            results = model.predict(source=img, save=False, conf=0.4)
 
             detections = []
             for r in results:
@@ -226,10 +269,10 @@ async def predict(image: UploadFile = File(...)):
                         "confidence": round(conf, 2),
                         "treatment": treatment,
                     })
-            
-            print(f"Detections for {filename}: {detections}")
 
             if not detections:
+                log_details = json.dumps({"pest": "None", "confidence": 0.0, "filename": image.filename})
+                log_audit("anonymous", "/predict", "success", log_details, ip)
                 return {
                     "status": "success",
                     "pest_detected": "None",
@@ -238,6 +281,13 @@ async def predict(image: UploadFile = File(...)):
                     "message": "No pests detected",
                 }
 
+            log_details = json.dumps({
+                "pest": detections[0]["pest_detected"],
+                "confidence": detections[0]["confidence"],
+                "filename": image.filename
+            })
+            log_audit("anonymous", "/predict", "success", log_details, ip)
+
             return {
                 "status": "success",
                 "pest_detected": detections[0]["pest_detected"],
@@ -247,8 +297,11 @@ async def predict(image: UploadFile = File(...)):
                 "all_detections": detections,
             }
         except Exception as e:
+            log_audit("anonymous", "/predict", "error", f"YOLO Inference error: {str(e)}", ip)
             raise HTTPException(status_code=500, detail=f"Inference error: {str(e)}")
     else:
+        log_details = json.dumps({"pest": "Aphids (Mock)", "confidence": 0.95, "filename": image.filename})
+        log_audit("anonymous", "/predict", "success", log_details, ip)
         return {
             "status": "success",
             "pest_detected": "Aphids (Mock)",
@@ -259,6 +312,5 @@ async def predict(image: UploadFile = File(...)):
 
 if __name__ == "__main__":
     import uvicorn
-
     port = int(os.environ.get("PORT", 5000))
     uvicorn.run(app, host="0.0.0.0", port=port)

gradle.properties

@@ -20,4 +20,7 @@ kotlin.code.style=official
 # Enables namespacing of each library's R class so that its R class includes only the
 # resources declared in the library itself and none from the library's dependencies,
 # thereby reducing the size of the R class for that library
-android.nonTransitiveRClass=true
+android.nonTransitiveRClass=true
+
+# Java Home configuration for Gradle build
+org.gradle.java.home=C:\\Program Files\\Android\\Android Studio\\jbr

gradlew.bat

@@ -39,6 +39,14 @@ for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
 set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
 
 @rem Find java.exe
+if not defined JAVA_HOME (
+    if exist "C:\Program Files\Android\Android Studio\jbr" (
+        set "JAVA_HOME=C:\Program Files\Android\Android Studio\jbr"
+    ) else if exist "C:\Program Files\Android\Android Studio\jre" (
+        set "JAVA_HOME=C:\Program Files\Android\Android Studio\jre"
+    )
+)
+
 if defined JAVA_HOME goto findJavaFromJavaHome
 
 set JAVA_EXE=java.exe