Update app.py

app.py

@@ -4,13 +4,26 @@ from openai import OpenAI
 
 print("Gradio version:", gr.__version__)
 
+# ===================== CONFIG =====================
 MODEL_HUMANIZER = "gpt-4.1-mini"
 MODEL_DETECTOR  = "gpt-4.1-mini"
 TEMPERATURE_REWRITE = 0.6
 MAX_TOKENS_REWRITE  = 1200
 
+STYLE_GUIDANCE = {
+    "Neutral":
+        "Tone: neutral, balanced, human. Keep meaning. Mix sentence lengths. Avoid robotic cadence & clichés.",
+    "Academic (Nature/IEEE)":
+        "Tone: formal, precise, journal-like (Nature/IEEE). Use hedging where appropriate; maintain terminology; avoid hype; keep citations intact.",
+    "Corporate (Professional Email)":
+        "Tone: concise, courteous, professional. Clear requests and actions; avoid slang; keep paragraphs short.",
+    "WhatsApp (Casual)":
+        "Tone: casual, conversational, natural texting vibe. Use contractions and light emojis sparingly; keep it friendly, not sloppy."
+}
+
+# ===================== OPENAI CLIENT =====================
 def get_client():
-    api_key  = os.environ.get("OPENAI_API_KEY")
+    api_key = os.environ.get("OPENAI_API_KEY")
     base_url = os.environ.get("OPENAI_BASE_URL")
     if not api_key:
         raise RuntimeError("Missing OPENAI_API_KEY")
@@ -18,41 +31,57 @@ def get_client():
         return OpenAI(api_key=api_key, base_url=base_url)
     return OpenAI(api_key=api_key)
 
-SYSTEM_HUMANIZER = (
-    "Rewrite text into neutral, balanced human English. Preserve meaning; improve clarity; "
-    "avoid robotic cadence; keep citations."
+# ===================== PROMPTS =====================
+SYSTEM_HUMANIZER_BASE = (
+    "You are Zawiyah AI Humanizer. Rewrite text into clear, human-sounding English. "
+    "Preserve meaning, data, names and citations. Improve clarity, reduce redundancy, vary sentence length, "
+    "avoid robotic cadence and generic AI phrasing. Do not invent facts."
+)
+
+USER_HUMANIZER_TEMPLATE = (
+    "Apply this style guidance: {style}\n\n"
+    "Rewrite the text naturally while keeping formatting (headings, bullets) tidy:\n"
+    "```text\n{src}\n```"
 )
-USER_HUMANIZER_TEMPLATE = "Rewrite this text:\n```text\n{src}\n```"
 
 SYSTEM_DETECTOR = (
-    "You are an AI-likeness detector. Return STRICT JSON: "
-    '{ "ai_likeness_score": float, "verdict": "Likely Human"|"Likely AI"|"Unclear", '
-    '"rationale": string }.'
+    "You are an AI-likeness detector. Return STRICT JSON only:\n"
+    '{ "ai_likeness_score": float,  "verdict": "Likely Human"|"Likely AI"|"Unclear", '
+    '"rationale": "1–2 concise reasons" }'
 )
+
 USER_DETECTOR_TEMPLATE = (
-    "ORIGINAL:\n```text\n{orig}\n```\n"
+    "ORIGINAL:\n```text\n{orig}\n```\n\n"
     "COMPARISON:\n```text\n{rew}\n```\n"
     "Return JSON only."
 )
 
-def humanize_text(text):
+# ===================== CORE FUNCS =====================
+def humanize_text(text: str, style_choice: str) -> str:
     try:
+        if not text or not text.strip():
+            return "Please paste some text."
+        style_note = STYLE_GUIDANCE.get(style_choice or "Neutral", STYLE_GUIDANCE["Neutral"])
         client = get_client()
         resp = client.chat.completions.create(
             model=MODEL_HUMANIZER,
             temperature=TEMPERATURE_REWRITE,
             max_tokens=MAX_TOKENS_REWRITE,
             messages=[
-                {"role": "system", "content": SYSTEM_HUMANIZER},
-                {"role": "user",   "content": USER_HUMANIZER_TEMPLATE.format(src=text)}
+                {"role": "system", "content": SYSTEM_HUMANIZER_BASE},
+                {"role": "user",   "content": USER_HUMANIZER_TEMPLATE.format(style=style_note, src=text)}
             ],
         )
         return resp.choices[0].message.content.strip()
     except Exception as e:
+        print("---- HUMANIZER ERROR ----")
+        print(traceback.format_exc())
         return f"ERROR: {e}"
 
-def detect_ai_likeness(original_text, comparison_text):
+def detect_ai_likeness(original_text: str, comparison_text: str) -> str:
     try:
+        if not comparison_text or not comparison_text.strip():
+            return "AI Likelihood: N/A\nVerdict: Unclear\nReason: No comparison text provided."
         client = get_client()
         resp = client.chat.completions.create(
             model=MODEL_DETECTOR,
@@ -60,26 +89,56 @@ def detect_ai_likeness(original_text, comparison_text):
             max_tokens=400,
             messages=[
                 {"role": "system", "content": SYSTEM_DETECTOR},
-                {"role": "user",   "content": USER_DETECTOR_TEMPLATE.format(orig=original_text, rew=comparison_text)}
+                {"role": "user",   "content": USER_DETECTOR_TEMPLATE.format(
+                    orig=original_text or "", rew=comparison_text or ""
+                )},
             ],
         )
         raw = resp.choices[0].message.content.strip()
-        data = json.loads(raw)
-        pct = float(data["ai_likeness_score"])*100
-        return f"AI Likelihood: {pct:.1f}%\nVerdict: {data['verdict']}\nReason: {data['rationale']}"
+        try:
+            data = json.loads(raw)
+            pct = float(data.get("ai_likeness_score", 0.5)) * 100
+            verdict = data.get("verdict", "Unclear")
+            reason  = data.get("rationale", "")
+            return f"AI Likelihood: {pct:.1f}%\nVerdict: {verdict}\nReason: {reason}"
+        except Exception:
+            return raw
     except Exception as e:
+        print("---- DETECTOR ERROR ----")
+        print(traceback.format_exc())
         return f"ERROR: {e}"
 
+# ===================== UI / THEME =====================
 custom_css = """
-#title {text-align:center; font-size:32px; font-weight:800;}
-.card {background:white; border:1px solid #eaeaea; border-radius:16px; padding:18px;
-       box-shadow:0 6px 18px rgba(0,0,0,0.06);}
+:root { --brand:#8E0000; }
+#title { text-align:center; font-size:32px; font-weight:800; }
+.gradio-container { max-width: 1000px !important; margin: auto; }
+
+/* Cards & toolbars */
+.card { background: #fff; border: 1px solid #eaeaea; border-radius: 16px; padding: 18px;
+        box-shadow: 0 6px 18px rgba(0,0,0,0.06); }
+.toolbar { background: #fff; border: 1px solid #eaeaea; border-radius: 12px; padding: 10px 14px;
+           box-shadow: 0 4px 14px rgba(0,0,0,0.05); }
+
+/* Dark mode via [data-theme="dark"] */
+:root[data-theme="dark"] body,
+:root[data-theme="dark"] .gradio-container { background: #0b0b0f !important; color: #eaeaea; }
+:root[data-theme="dark"] .card { background: #101116; border-color: #222; box-shadow: 0 6px 18px rgba(0,0,0,0.5); }
+:root[data-theme="dark"] .toolbar { background: #101116; border-color: #222; box-shadow: 0 4px 14px rgba(0,0,0,0.4); }
+:root[data-theme="dark"] textarea, 
+:root[data-theme="dark"] input, 
+:root[data-theme="dark"] .gr-textbox textarea { background: #151723 !important; color: #f0f0f0 !important; border-color: #2a2d3a !important; }
+:root[data-theme="dark"] .gr-button { border-color: #2a2d3a !important; }
+
+/* Footer */
+#footer { text-align:center; font-size:12px; color:#666; margin-top:12px; }
+:root[data-theme="dark"] #footer { color:#aaa; }
 """
 
 theme = gr.themes.Soft(primary_hue="red", neutral_hue="gray")
 
 with gr.Blocks(theme=theme, css=custom_css) as demo:
-
+    # Title
     gr.Markdown("""
     <div id="title">🕌 Zawiyah AI Collective — <span style='color:#8E0000'>Humanizer AI</span></div>
     <p style="text-align:center; font-size:16px; margin-top:-8px">
@@ -87,39 +146,86 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
     </p><hr>
     """)
 
+    # Far-right small toolbar with Dark Mode toggle (P2)
+    with gr.Row():
+        gr.Markdown("")  # spacer (left)
+        with gr.Row(elem_classes="toolbar"):
+            dark_btn = gr.Button(value="🌙 Dark mode", elem_id="dark_toggle")
+
+    # Top toolbar (full width under title) — Writing Style dropdown
+    with gr.Row(elem_classes="toolbar"):
+        gr.Markdown("**Writing Style**")
+        style_dropdown = gr.Dropdown(
+            choices=["Neutral", "Academic (Nature/IEEE)", "Corporate (Professional Email)", "WhatsApp (Casual)"],
+            value="Neutral",
+            interactive=True,
+            label=None
+        )
+
+    # Input + actions
     with gr.Column(elem_classes="card"):
-        input_text = gr.Textbox(label="✍️ Paste your text", lines=10)
+        input_text = gr.Textbox(label="✍️ Paste Your Text Here", placeholder="Enter text…", lines=10)
         with gr.Row():
             btn_humanize = gr.Button("🧠 Humanize Text")
             btn_detect   = gr.Button("🕵️ AI Likelihood Detector")
 
+    # Results
     with gr.Row():
         with gr.Column(elem_classes="card"):
-            gr.Markdown("### Humanized Output")
+            gr.Markdown("### ✅ Humanized Output")
             output_refined = gr.Textbox(show_label=False, lines=14)
             copy_btn = gr.Button("📋 Copy")
         with gr.Column(elem_classes="card"):
-            gr.Markdown("### AI Likelihood Result")
-            detector_result = gr.Textbox(show_label=False, lines=8)
+            gr.Markdown("### 📊 AI Likelihood Result")
+            detector_result = gr.Textbox(show_label=False, lines=8, interactive=False)
 
-    btn_humanize.click(humanize_text, input_text, output_refined)
+    # Wiring — core actions
+    btn_humanize.click(humanize_text, [input_text, style_dropdown], output_refined)
     btn_detect.click(detect_ai_likeness, [input_text, output_refined], detector_result)
 
+    # Copy to clipboard (Gradio 5 uses js= and a dummy python fn)
     copy_btn.click(
         lambda x: x,
         inputs=output_refined,
         outputs=None,
         js="""
         function(text){
-            navigator.clipboard.writeText(text);
-            alert("Copied ✅");
+            if(!text){ alert("Nothing to copy"); return; }
+            navigator.clipboard.writeText(text)
+              .then(()=> alert("Copied ✅"))
+              .catch(()=> alert("Copy failed"));
         }
         """
     )
 
+    # Dark mode toggle (Gradio 5: js= to flip documentElement dataset theme)
+    # Also swap button text between 🌙 Dark mode / ☀️ Light mode
+    dark_btn.click(
+        lambda: None,
+        inputs=None,
+        outputs=None,
+        js="""
+        function(){
+          const root = document.documentElement;
+          const current = root.getAttribute('data-theme');
+          const next = current === 'dark' ? '' : 'dark';
+          root.setAttribute('data-theme', next);
+
+          // flip button label
+          const btn = document.getElementById('dark_toggle');
+          if (btn){
+            btn.innerText = (next === 'dark') ? '☀️ Light mode' : '🌙 Dark mode';
+          }
+        }
+        """
+    )
+
+    gr.Markdown("<div id='footer'>Zawiyah AI Collective · Malaysia · v1.0 BETA · Built with Gradio</div>")
+
 demo.queue().launch()
 
 
 
 
 
+