Upload 6 files

app/config.py

@@ -1,5 +1,5 @@
 """
-إعدادات التطبيق - تُقرأ من متغيرات البيئة (.env)
+All application settings — read from environment variables or .env file.
 """
 from functools import lru_cache
 
@@ -7,24 +7,25 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
 class Settings(BaseSettings):
-    # إعدادات عامة
-    app_name: str = "HF Chat Backend"
+    app_name: str = "AI Tools Backend"
     app_env: str = "development"
     debug: bool = False
 
-    # إعدادات Hugging Face
+    # Auth — comma-separated API keys
+    api_keys: str = ""
+
+    # File management
+    upload_dir: str = "./uploads"
+    output_dir: str = "./outputs"
+    max_upload_size_mb: int = 100
+
+    # AI / LLM (optional — for chat proxy)
     hf_api_token: str = ""
     hf_model_id: str = "meta-llama/Llama-3.1-8B-Instruct"
-    hf_provider: str | None = None  # مثال: "hf-inference" أو "together" أو "fireworks-ai"
-    hf_timeout_seconds: int = 60
+    hf_provider: str | None = None
 
-    # إعدادات توليد النص الافتراضية
-    default_max_new_tokens: int = 512
-    default_temperature: float = 0.7
-    default_top_p: float = 0.95
-
-    # أمان بسيط: مفتاح API اختياري لحماية الباكند نفسه
-    backend_api_key: str = ""
+    # Security tools toggle
+    pentest_enabled: bool = True
 
     # CORS
     cors_allow_origins: str = "*"
@@ -36,6 +37,10 @@ class Settings(BaseSettings):
         extra="ignore",
     )
 
+    @property
+    def api_key_list(self) -> list[str]:
+        return [k.strip() for k in self.api_keys.split(",") if k.strip()]
+
 
 @lru_cache
 def get_settings() -> Settings:

app/exceptions.py

@@ -0,0 +1,79 @@
+"""
+Custom exceptions + global error handlers.
+"""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from fastapi import FastAPI, Request, status
+from fastapi.responses import JSONResponse
+
+from app.schemas import ErrorResponse
+
+logger = logging.getLogger("ai_tools.errors")
+
+
+class ToolError(Exception):
+    """Base error for tool execution failures."""
+
+    def __init__(
+        self,
+        tool_name: str,
+        message: str,
+        status_code: int = 400,
+        detail: Any = None,
+    ) -> None:
+        self.tool_name = tool_name
+        self.message = message
+        self.status_code = status_code
+        self.detail = detail
+
+
+class FileTooLargeError(ToolError):
+    def __init__(self, tool_name: str, max_mb: int) -> None:
+        super().__init__(tool_name, f"File exceeds maximum size of {max_mb}MB", 413)
+
+
+class UnsupportedFormatError(ToolError):
+    def __init__(self, tool_name: str, fmt: str) -> None:
+        super().__init__(tool_name, f"Unsupported format: {fmt}", 400)
+
+
+class SecurityToolDisabledError(ToolError):
+    def __init__(self, tool_name: str) -> None:
+        super().__init__(tool_name, "Security tools are disabled (pentest_enabled=false)", 403)
+
+
+class FileNotFoundError(ToolError):
+    def __init__(self, tool_name: str, file_id: str) -> None:
+        super().__init__(tool_name, f"File not found: {file_id}", 404)
+
+
+class UnauthorizedError(ToolError):
+    def __init__(self) -> None:
+        super().__init__("auth", "Invalid or missing API key", 401)
+
+
+def register_error_handlers(app: FastAPI) -> None:
+    @app.exception_handler(ToolError)
+    async def tool_error_handler(request: Request, exc: ToolError) -> JSONResponse:
+        logger.warning("Tool error [%s]: %s", exc.tool_name, exc.message)
+        return JSONResponse(
+            status_code=exc.status_code,
+            content=ErrorResponse(
+                error=exc.tool_name,
+                detail=exc.message,
+            ).model_dump(),
+        )
+
+    @app.exception_handler(Exception)
+    async def unhandled_handler(request: Request, exc: Exception) -> JSONResponse:
+        logger.exception("Unhandled exception on %s %s", request.method, request.url.path)
+        return JSONResponse(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            content=ErrorResponse(
+                error="internal_server_error",
+                detail=str(exc),
+            ).model_dump(),
+        )

app/logging_config.py

@@ -1,5 +1,5 @@
 """
-إعداد Logging مهيكل (Structured Logging) بصيغة JSON.
+Structured JSON logging configuration.
 """
 import json
 import logging
@@ -24,10 +24,9 @@ def configure_logging(debug: bool = False) -> None:
     handler = logging.StreamHandler(sys.stdout)
     handler.setFormatter(JsonFormatter())
 
-    root_logger = logging.getLogger()
-    root_logger.handlers = [handler]
-    root_logger.setLevel(logging.DEBUG if debug else logging.INFO)
+    root = logging.getLogger()
+    root.handlers = [handler]
+    root.setLevel(logging.DEBUG if debug else logging.INFO)
 
-    # تهدئة بعض المكتبات الثرثارة
     logging.getLogger("httpx").setLevel(logging.WARNING)
-    logging.getLogger("huggingface_hub").setLevel(logging.INFO)
+    logging.getLogger("httpcore").setLevel(logging.WARNING)

app/main.py

@@ -1,51 +1,52 @@
 """
-نقطة الدخول الرئيسية لتطبيق FastAPI.
+AI Tools Backend — Application factory and entry point.
 """
 import logging
 from contextlib import asynccontextmanager
 
-from fastapi import FastAPI, Request, status
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
 
+from app.auth.api_key import verify_api_key
 from app.config import get_settings
+from app.core.file_manager import FileManager
+from app.core.plugin_manager import plugin_manager
+from app.exceptions import register_error_handlers
 from app.logging_config import configure_logging
-from app.routers import chat, health
-from app.schemas import ErrorResponse
+from app.routers import ai_routes, document_routes, health_routes, media_routes, security_routes, tools_routes
 
 settings = get_settings()
 configure_logging(debug=settings.debug)
-logger = logging.getLogger("hf_chat_backend.main")
+logger = logging.getLogger("ai_tools.main")
+
+# Global file manager instance
+file_manager = FileManager(settings)
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    logger.info(
-        "Starting %s [env=%s] using model=%s",
-        settings.app_name,
-        settings.app_env,
-        settings.hf_model_id,
-    )
+    logger.info("Starting %s", settings.app_name)
+    plugin_manager.auto_discover()
+    logger.info("Loaded %d tools", plugin_manager.count)
     yield
     logger.info("Shutting down %s", settings.app_name)
 
 
 app = FastAPI(
     title=settings.app_name,
-    description="باكند FastAPI بسيط لاستدعاء نموذج LLM مستضاف على Hugging Face",
+    description="AI Tools Backend — 47+ tools for documents, media, security, and AI",
     version="1.0.0",
     docs_url="/docs",
     redoc_url="/redoc",
-    openapi_url="/openapi.json",
     lifespan=lifespan,
 )
 
+# CORS
 origins = (
     ["*"]
     if settings.cors_allow_origins.strip() == "*"
     else [o.strip() for o in settings.cors_allow_origins.split(",") if o.strip()]
 )
-
 app.add_middleware(
     CORSMiddleware,
     allow_origins=origins,
@@ -54,15 +55,12 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+register_error_handlers(app)
 
-@app.exception_handler(Exception)
-async def unhandled_exception_handler(request: Request, exc: Exception) -> JSONResponse:
-    logger.exception("Unhandled exception on %s %s", request.method, request.url.path)
-    return JSONResponse(
-        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-        content=ErrorResponse(error="internal_server_error", detail=str(exc)).model_dump(),
-    )
-
-
-app.include_router(health.router)
-app.include_router(chat.router)
+# Register routers
+app.include_router(health_routes.router)
+app.include_router(tools_routes.router)
+app.include_router(document_routes.router)
+app.include_router(media_routes.router)
+app.include_router(security_routes.router)
+app.include_router(ai_routes.router)

app/schemas.py

@@ -1,41 +1,56 @@
 """
-Pydantic Schemas الخاصة بطلبات/ردود الدردشة
+Shared Pydantic schemas for all tool inputs/outputs.
 """
-from typing import Literal
-
-from pydantic import BaseModel, Field
+from __future__ import annotations
 
+from typing import Any, Literal
 
-class ChatMessage(BaseModel):
-    role: Literal["system", "user", "assistant"]
-    content: str = Field(..., min_length=1)
+from pydantic import BaseModel, Field
 
 
-class ChatRequest(BaseModel):
-    messages: list[ChatMessage] = Field(
-        ..., min_length=1, description="سجل المحادثة كاملاً"
-    )
-    max_new_tokens: int | None = Field(
-        default=None, ge=1, le=4096, description="أقصى عدد Tokens للرد"
-    )
-    temperature: float | None = Field(default=None, ge=0.0, le=2.0)
-    top_p: float | None = Field(default=None, ge=0.0, le=1.0)
-    stream: bool = Field(default=False, description="تفعيل البث اللحظي (SSE)")
+class ToolResult(BaseModel):
+    """Unified response from any tool execution."""
+    success: bool
+    tool: str
+    result: Any = None
+    error: str | None = None
+    file_id: str | None = None
+    file_name: str | None = None
 
 
-class ChatResponse(BaseModel):
-    model: str
-    reply: str
-    finish_reason: str | None = None
+class ErrorResponse(BaseModel):
+    error: str
+    detail: str | None = None
 
 
 class HealthResponse(BaseModel):
     status: Literal["ok", "degraded"]
     app_name: str
-    app_env: str
-    model: str
+    version: str
+    tools_count: int
 
 
-class ErrorResponse(BaseModel):
-    error: str
-    detail: str | None = None
+class ToolInfo(BaseModel):
+    name: str
+    description: str
+    category: str
+    input_schema: dict[str, Any]
+
+
+class ToolListResponse(BaseModel):
+    tools: list[ToolInfo]
+    total: int
+
+
+class FileUploadResponse(BaseModel):
+    file_id: str
+    filename: str
+    size: int
+    content_type: str
+
+
+class TaskResponse(BaseModel):
+    task_id: str
+    status: Literal["pending", "running", "completed", "failed"]
+    result: Any = None
+    error: str | None = None