Update app.py

app.py

@@ -5,70 +5,84 @@ from llama_cpp import Llama
 from contextlib import asynccontextmanager
 from huggingface_hub import hf_hub_download
 
-
-# =========================
-# MODEL CONFIG
-# =========================
+# ==================================================
+# MODEL CONFIGURATION
+# ==================================================
 MODEL_REPO = "bartowski/Phi-3.1-mini-4k-instruct-GGUF"
 MODEL_FILE = "Phi-3.1-mini-4k-instruct-IQ2_M.gguf"
 
-MODEL_PATH = hf_hub_download(
-    repo_id=MODEL_REPO,
-    filename=MODEL_FILE,
-)
-
-
 N_THREADS = int(os.getenv("N_THREADS", "8"))
 N_CTX = int(os.getenv("N_CTX", "2048"))
 N_BATCH = int(os.getenv("N_BATCH", "256"))
 
 llm = None
 
-# =========================
-# FASTAPI LIFESPAN
-# =========================
+# ==================================================
+# FASTAPI LIFESPAN (MODEL LOAD)
+# ==================================================
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     global llm
+
+    model_path = hf_hub_download(
+        repo_id=MODEL_REPO,
+        filename=MODEL_FILE,
+    )
+
     llm = Llama(
-    model_path=MODEL_PATH,
-    n_ctx=N_CTX,
-    n_threads=N_THREADS,
-    n_batch=N_BATCH,
-    verbose=False,
+        model_path=model_path,
+        n_ctx=N_CTX,
+        n_threads=N_THREADS,
+        n_batch=N_BATCH,
+        verbose=False,
     )
 
     yield
 
-app = FastAPI(title="AI Humanizer", lifespan=lifespan)
+# ==================================================
+# APP INIT
+# ==================================================
+app = FastAPI(
+    title="AI Humanizer",
+    description="Academic-safe AI Humanizer",
+    version="1.0",
+    lifespan=lifespan
+)
+
+# ==================================================
+# DUMMY AUTH VERIFY (Frontend Fix)
+# ==================================================
+@app.get("/api/auth/verify")
+def verify_auth():
+    return {"authenticated": True}
 
-# =========================
+# ==================================================
 # INPUT SCHEMA
-# =========================
+# ==================================================
 class HumanizeRequest(BaseModel):
     text: str
     section: str  # abstract | introduction | methodology | results | discussion
     author_notes: str | None = None
 
-# =========================
-# SECTION-AWARE STYLE MAP
-# =========================
+# ==================================================
+# SECTION STYLE MAP
+# ==================================================
 SECTION_STYLE = {
-    "abstract": "Write concisely, densely, and objectively. Avoid narrative tone.",
-    "introduction": "Use contextual and motivational academic tone. Explain relevance.",
-    "methodology": "Be procedural, precise, and restrained. No persuasion.",
-    "results": "Be cautious, numerical, and observational. No strong claims.",
+    "abstract": "Write concisely and densely with objective academic tone.",
+    "introduction": "Use contextual, motivational academic tone.",
+    "methodology": "Be procedural, precise, and restrained.",
+    "results": "Be cautious, numerical, and observational.",
     "discussion": "Be interpretive, reflective, and analytical."
 }
 
-# =========================
+# ==================================================
 # PROMPT BUILDER
-# =========================
+# ==================================================
 def build_prompt(text: str, section: str, author_notes: str | None):
     style = SECTION_STYLE.get(section.lower(), "Use formal academic tone.")
 
     notes_block = (
-        f"\nAuthor context (do not invent new reasoning):\n{author_notes}\n"
+        f"\nAuthor context (do not invent reasoning):\n{author_notes}\n"
         if author_notes else ""
     )
 
@@ -93,7 +107,7 @@ HUMANIZATION RULES:
 - Reorder clauses where appropriate
 - Avoid overused academic fillers (e.g., "Moreover", "Furthermore", "It is important to note")
 - Prefer implicit transitions over explicit signposting
-- Preserve and highlight existing author reasoning or constraints
+- Preserve existing author reasoning and constraints
 - Use controlled lexical variation without replacing technical terms
 
 {notes_block}
@@ -108,9 +122,9 @@ OUTPUT:
 Return ONLY the humanized text.
 """.strip()
 
-# =========================
-# API ENDPOINT
-# =========================
+# ==================================================
+# HUMANIZE ENDPOINT
+# ==================================================
 @app.post("/humanize")
 def humanize(req: HumanizeRequest):
     if not req.text.strip():
@@ -124,8 +138,8 @@ def humanize(req: HumanizeRequest):
         temperature=0.4,
         top_p=0.9,
         repetition_penalty=1.1,
-        stop=["<|end|>"]
     )
 
-    output = response["choices"][0]["text"].strip()
-    return {"humanized_text": output}
+    return {
+        "humanized_text": response["choices"][0]["text"].strip()
+    }