Updated

Dockerfile

@@ -3,7 +3,7 @@
 # ==============================================================
 
 # Use lightweight official Python image
-FROM python:3.11-slim
+FROM python:3.10-slim
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
@@ -28,7 +28,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # Expose port for FastAPI
-EXPOSE 8000
+EXPOSE 7860
 
 # Run the FastAPI app with Uvicorn
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,5 +1,5 @@
 # ==============================================================
-# Tech Disciples AI Backend — Optimized Stable Release
+# Tech Disciples AI Backend — Character-Controlled Release
 # ==============================================================
 
 import os
@@ -10,24 +10,27 @@ from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from transformers import pipeline
 
-# ==============================================================
+# --------------------------------------------------------------
 # Logging Setup
-# ==============================================================
-logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
+# --------------------------------------------------------------
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s"
+)
 logger = logging.getLogger("Tech Disciples AI")
 
-# ==============================================================
+# --------------------------------------------------------------
 # FastAPI App Initialization
-# ==============================================================
+# --------------------------------------------------------------
 app = FastAPI(title="Tech Disciples AI")
 
 @app.get("/")
 async def root():
-    return {"status": "Tech Disciples AI Backend is running and stable."}
+    return {"status": "✅ Tech Disciples AI Backend is running and stable."}
 
-# ==============================================================
-# Authentication Configuration
-# ==============================================================
+# --------------------------------------------------------------
+# Authentication
+# --------------------------------------------------------------
 PROJECT_API_KEY = os.getenv("PROJECT_API_KEY", "techdisciplesai404")
 
 def check_auth(authorization: str | None):
@@ -40,23 +43,24 @@ def check_auth(authorization: str | None):
     if token != PROJECT_API_KEY:
         raise HTTPException(status_code=403, detail="Invalid token")
 
-# ==============================================================
+# --------------------------------------------------------------
 # Global Exception Handler
-# ==============================================================
+# --------------------------------------------------------------
 @app.exception_handler(Exception)
 async def global_exception_handler(request: Request, exc: Exception):
     logger.error(f"Unhandled error: {exc}")
     return JSONResponse(status_code=500, content={"error": str(exc)})
 
-# ==============================================================
-# Request Models
-# ==============================================================
+# --------------------------------------------------------------
+# Request Model
+# --------------------------------------------------------------
 class ChatRequest(BaseModel):
     query: str
+    tone: str | None = "neutral"  # Optional tone parameter
 
-# ==============================================================
+# --------------------------------------------------------------
 # Hugging Face Configuration
-# ==============================================================
+# --------------------------------------------------------------
 HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 if not HF_TOKEN:
@@ -64,13 +68,13 @@ if not HF_TOKEN:
 else:
     logger.info("✅ Hugging Face token detected.")
 
-# Device selection
+# Device setup
 device = 0 if torch.cuda.is_available() else -1
 logger.info(f"🧠 Using device: {'GPU' if device == 0 else 'CPU'}")
 
-# ==============================================================
+# --------------------------------------------------------------
 # Model Pipeline Initialization
-# ==============================================================
+# --------------------------------------------------------------
 try:
     chat_pipe = pipeline(
         "text-generation",
@@ -83,32 +87,82 @@ except Exception as e:
     chat_pipe = None
     logger.error(f"❌ Failed to load model pipeline: {e}")
 
-# ==============================================================
-# Helper Functions
-# ==============================================================
-def run_conversational(pipe, prompt: str) -> str:
-    """Executes text generation safely and returns formatted output."""
+# --------------------------------------------------------------
+# System / Character Template
+# --------------------------------------------------------------
+SYSTEM_TEMPLATE = """
+You are Tech Disciples AI — an intelligent, respectful, and spiritually aware assistant 
+created to provide biblical insight, wisdom, and intelligent guidance.
+
+Your core personality:
+- Speak with humility and warmth.
+- Use clear, simple, and sincere language.
+- When asked about faith, respond with biblical understanding and gentleness.
+- When asked about general or technical topics, respond with calm intelligence.
+- Avoid arguments, controversy, or judgmental language.
+- Always sound like a wise, grounded, caring mentor.
+"""
+
+# --------------------------------------------------------------
+# Helper Function
+# --------------------------------------------------------------
+def run_conversational(pipe, user_input: str, tone: str = "neutral") -> str:
+    """Executes text generation safely and returns formatted output with personality control."""
     if not pipe:
         return "⚠️ Model pipeline not initialized."
 
+    # Define tone modifiers
+    tone_instruction = {
+        "neutral": "Keep tone calm, balanced, and thoughtful.",
+        "friendly": "Use a warm, conversational tone with kindness and empathy.",
+        "motivational": "Speak with encouragement, confidence, and faith-driven energy.",
+        "scholarly": "Respond with logical, scripture-supported, and analytical reasoning.",
+        "pastoral": "Speak gently as a guide offering spiritual comfort and care.",
+    }.get(tone.lower(), "Keep tone calm and thoughtful.")
+
     try:
-        output = pipe(prompt, max_new_tokens=200, temperature=0.3, do_sample=True)
-        if isinstance(output, list) and output:
-            return output[0].get("generated_text", str(output))
+        # Combine the system and tone template with the user input
+        prompt = (
+            f"{SYSTEM_TEMPLATE}\n"
+            f"{tone_instruction}\n\n"
+            f"User: {user_input}\n"
+            f"Tech Disciples AI:"
+        )
+
+        output = pipe(
+            prompt,
+            max_new_tokens=220,
+            temperature=0.4,
+            top_p=0.9,
+            repetition_penalty=1.1,
+            do_sample=True,
+        )
+
+        # Parse generated text
+        if isinstance(output, list) and len(output) > 0:
+            text = output[0].get("generated_text", str(output))
+            if "Tech Disciples AI:" in text:
+                text = text.split("Tech Disciples AI:")[-1].strip()
+            return text
+        
         return str(output)
+    
     except Exception as e:
         logger.error(f"Conversational pipeline error: {e}")
-        return f"⚠️ Model error: {e}"
+        return f"⚠️ Model error: {str(e)}"
 
-# ==============================================================
+# --------------------------------------------------------------
 # API Endpoints
-# ==============================================================
+# --------------------------------------------------------------
 @app.post("/ai-chat")
 async def ai_chat(req: ChatRequest, authorization: str | None = Header(None)):
-    """Handles conversational AI chat requests."""
+    """
+    Main conversational endpoint for Tech Disciples AI.
+    Accepts 'query' and optional 'tone' parameters.
+    """
     check_auth(authorization)
-    reply = run_conversational(chat_pipe, req.query)
-    return {"reply": reply}
+    reply = run_conversational(chat_pipe, req.query, req.tone)
+    return {"reply": reply, "tone_used": req.tone}
 
 # ==============================================================
 # END OF FILE