Add AI Text Category Suggestion

backend/config.py

@@ -27,6 +27,7 @@ class Config:
     # API Keys
     gemini_api_key: str
     telegram_bot_token: str
+    hf_token: Optional[str]
 
     # Database
     database_url: str
@@ -66,6 +67,8 @@ def from_env(cls) -> "Config":
         if not telegram_bot_token:
             errors.append("TELEGRAM_BOT_TOKEN is required")
 
+        hf_token = os.getenv("HF_TOKEN")
+
         # Database with default
         database_url = os.getenv(
             "DATABASE_URL", 
@@ -121,6 +124,7 @@ def from_env(cls) -> "Config":
         return cls(
             gemini_api_key=gemini_api_key,
             telegram_bot_token=telegram_bot_token,
+            hf_token=hf_token,
             database_url=database_url,
             environment=environment,
             debug=debug,
@@ -274,6 +278,11 @@ def get_telegram_bot_token() -> str:
     return get_config().telegram_bot_token
 
 
+def get_hf_token() -> Optional[str]:
+    """Get Hugging Face token from config."""
+    return get_config().hf_token
+
+
 def get_database_url() -> str:
     """Get database URL from config."""
     return get_config().database_url

backend/hf_api_service.py

@@ -33,6 +33,9 @@
 # Speech-to-Text Model (Whisper)
 WHISPER_API_URL = "https://router.huggingface.co/models/openai/whisper-large-v3-turbo"
 
+# Zero-Shot Text Classification Model
+ZERO_SHOT_TEXT_API_URL = "https://router.huggingface.co/models/facebook/bart-large-mnli"
+
 async def _make_request(client, url, payload):
     try:
         response = await client.post(url, headers=headers, json=payload, timeout=20.0)
@@ -456,3 +459,40 @@ async def detect_abandoned_vehicle_clip(image: Union[Image.Image, bytes], client
     labels = ["abandoned car", "rusted vehicle", "car with flat tires", "wrecked car", "normal parked car"]
     targets = ["abandoned car", "rusted vehicle", "car with flat tires", "wrecked car"]
     return await _detect_clip_generic(image, labels, targets, client)
+
+async def classify_text_category(text: str, client: httpx.AsyncClient = None):
+    """
+    Classifies text into civic issue categories using Zero-Shot Classification.
+    Returns the top category and confidence.
+    """
+    labels = [
+        "pothole", "garbage dump", "water leak", "broken street light",
+        "broken infrastructure", "traffic congestion", "fire accident",
+        "stray animal threat", "fallen tree hazard", "pest infestation",
+        "clean area", "noise pollution"
+    ]
+
+    payload = {
+        "inputs": text,
+        "parameters": {"candidate_labels": labels}
+    }
+
+    if client:
+        result = await _make_request(client, ZERO_SHOT_TEXT_API_URL, payload)
+    else:
+        async with httpx.AsyncClient() as new_client:
+            result = await _make_request(new_client, ZERO_SHOT_TEXT_API_URL, payload)
+
+    # Result format: {'sequence': '...', 'labels': ['pothole', ...], 'scores': [0.9, ...]}
+    if isinstance(result, dict) and 'labels' in result and 'scores' in result:
+        top_label = result['labels'][0]
+        top_score = result['scores'][0]
+
+        # Simple mapping to internal categories if needed, or return raw
+        return {
+            "category": top_label,
+            "confidence": top_score,
+            "all_scores": dict(zip(result['labels'][:3], result['scores'][:3]))
+        }
+
+    return {"category": "unknown", "confidence": 0}

backend/routers/analysis.py

@@ -1,8 +1,10 @@
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Request
 from pydantic import BaseModel
 from typing import List, Optional, Any
 
 from backend.priority_engine import priority_engine
+from backend.hf_api_service import classify_text_category
+from backend.dependencies import get_http_client
 
 router = APIRouter()
 
@@ -36,3 +38,22 @@ def analyze_issue(request: AnalyzeIssueRequest):
         suggested_categories=result["suggested_categories"],
         reasoning=result["reasoning"]
     )
+
+class CategorySuggestionRequest(BaseModel):
+    text: str
+
+@router.post("/api/suggest-category-text")
+async def suggest_category_text(request: Request, body: CategorySuggestionRequest):
+    """
+    Suggests a category based on the text description using Zero-Shot Classification.
+    """
+    if not body.text or len(body.text) < 5:
+        raise HTTPException(status_code=400, detail="Text must be at least 5 characters long")
+
+    try:
+        client = get_http_client(request)
+        result = await classify_text_category(body.text, client=client)
+        return result
+    except Exception as e:
+        # Fallback or error
+        raise HTTPException(status_code=500, detail=f"Classification service unavailable: {str(e)}")

backend/schemas.py

@@ -33,6 +33,13 @@ class ActionPlan(BaseModel):
 class ChatRequest(BaseModel):
     query: str = Field(..., min_length=1, max_length=1000, description="Chat query text")
 
+    @field_validator('query')
+    @classmethod
+    def validate_query(cls, v):
+        if not v.strip():
+            raise ValueError('Query cannot be empty or whitespace only')
+        return v.strip()
+
 class ChatResponse(BaseModel):
     response: str
 

frontend/src/App.jsx

@@ -23,22 +23,8 @@ const GrievanceView = React.lazy(() => import('./views/GrievanceView'));
 const NotFound = React.lazy(() => import('./views/NotFound'));
 
 // Lazy Load Detectors
-const PotholeDetector = React.lazy(() => import('./PotholeDetector'));
-const GarbageDetector = React.lazy(() => import('./GarbageDetector'));
-const VandalismDetector = React.lazy(() => import('./VandalismDetector'));
-const FloodDetector = React.lazy(() => import('./FloodDetector'));
-const InfrastructureDetector = React.lazy(() => import('./InfrastructureDetector'));
-const IllegalParkingDetector = React.lazy(() => import('./IllegalParkingDetector'));
-const StreetLightDetector = React.lazy(() => import('./StreetLightDetector'));
-const FireDetector = React.lazy(() => import('./FireDetector'));
-const StrayAnimalDetector = React.lazy(() => import('./StrayAnimalDetector'));
-const BlockedRoadDetector = React.lazy(() => import('./BlockedRoadDetector'));
-const TreeDetector = React.lazy(() => import('./TreeDetector'));
-const PestDetector = React.lazy(() => import('./PestDetector'));
 const SmartScanner = React.lazy(() => import('./SmartScanner'));
 const GrievanceAnalysis = React.lazy(() => import('./views/GrievanceAnalysis'));
-const NoiseDetector = React.lazy(() => import('./NoiseDetector'));
-const CivicEyeDetector = React.lazy(() => import('./CivicEyeDetector'));
 const CivicInsight = React.lazy(() => import('./views/CivicInsight'));
 const MyReportsView = React.lazy(() => import('./views/MyReportsView'));
 
@@ -296,52 +282,8 @@ function AppContent() {
                 }
               />
               <Route path="/verify/:id" element={<VerifyView />} />
-              <Route path="/pothole" element={<PotholeDetector onBack={() => navigate('/')} />} />
-              <Route path="/garbage" element={<GarbageDetector onBack={() => navigate('/')} />} />
-              <Route
-                path="/vandalism"
-                element={
-                  <div className="flex flex-col h-full">
-                    <button onClick={() => navigate('/')} className="self-start text-blue-600 mb-2">
-                      &larr; Back
-                    </button>
-                    <VandalismDetector />
-                  </div>
-                }
-              />
-              <Route
-                path="/flood"
-                element={
-                  <div className="flex flex-col h-full">
-                    <button onClick={() => navigate('/')} className="self-start text-blue-600 mb-2">
-                      &larr; Back
-                    </button>
-                    <FloodDetector />
-                  </div>
-                }
-              />
-              <Route
-                path="/infrastructure"
-                element={<InfrastructureDetector onBack={() => navigate('/')} />}
-              />
-              <Route path="/parking" element={<IllegalParkingDetector onBack={() => navigate('/')} />} />
-              <Route path="/streetlight" element={<StreetLightDetector onBack={() => navigate('/')} />} />
-              <Route path="/fire" element={<FireDetector onBack={() => navigate('/')} />} />
-              <Route path="/animal" element={<StrayAnimalDetector onBack={() => navigate('/')} />} />
-              <Route path="/blocked" element={<BlockedRoadDetector onBack={() => navigate('/')} />} />
-              <Route path="/tree" element={<TreeDetector onBack={() => navigate('/')} />} />
-              <Route path="/pest" element={<PestDetector onBack={() => navigate('/')} />} />
               <Route path="/smart-scan" element={<SmartScanner onBack={() => navigate('/')} />} />
               <Route path="/grievance-analysis" element={<GrievanceAnalysis onBack={() => navigate('/')} />} />
-              <Route path="/noise" element={<NoiseDetector onBack={() => navigate('/')} />} />
-              <Route path="/safety-check" element={
-                <div className="flex flex-col h-full p-4">
-                  <button onClick={() => navigate('/')} className="self-start text-blue-600 mb-2 font-bold">
-                    &larr; Back
-                  </button>
-                  <CivicEyeDetector onBack={() => navigate('/')} />
-                </div>
-              } />
               <Route path="/my-reports" element={
                 <ProtectedRoute>
                   <MyReportsView />