Update server/app.py

server/app.py

@@ -4,11 +4,13 @@ import uvicorn
 import numpy as np
 import gradio as gr
 from fastapi import FastAPI
+
+# Path fix taaki env.py mil jaye
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 from env import EmailTriageEnv, URGENCY_LABELS, ROUTING_LABELS, RESOLUTION_LABELS
 
 app = FastAPI()
 
-# --- Agent Logic ---
 def smart_agent_logic(desc):
     desc = desc.lower()
     if any(x in desc for x in ["password", "hacked", "breach", "phish", "threat", "ransomware"]):
@@ -19,13 +21,11 @@ def smart_agent_logic(desc):
         return [0, 1, 1]
     return [0, 0, 0]
 
-# --- Core Function ---
 def run_demo(task):
     try:
         env = EmailTriageEnv(task=task)
         env.reset()
-        results = []
-        total_reward = 0
+        results = []; total_reward = 0
         print(f"[START] Task: {task}")
         for i, email in enumerate(env._queue):
             action = smart_agent_logic(email['description'])
@@ -36,31 +36,32 @@ def run_demo(task):
             results.append(f"#{i+1} [{task.upper()}] {email['description'][:30]}... | {status}")
         score = total_reward / len(env._queue) if env._queue else 0
         print(f"[END] Final Score: {score}")
-        return "\n".join(results) + f"\n\n--- FINAL SCORE: {score:.3f} / 1.000 ---"
+        return "\n".join(results) + f"\n\n--- FINAL SCORE: {score:.3f} ---"
     except Exception as e:
         return f"Error: {str(e)}"
 
-# --- API for Validator ---
+# --- REQUIRED BY VALIDATOR ---
+def main():
+    print("--- 🚀 STARTING MULTI-MODE DEPLOYMENT TEST ---")
+    for level in ["easy", "medium", "hard"]:
+        print(run_demo(level))
+
 @app.post("/reset")
 async def reset_endpoint():
-    return {"status": "success", "message": "Environment Reset OK"}
-
-@app.get("/status")
-async def health_check():
-    return {"status": "online"}
+    return {"status": "success"}
 
-# --- Gradio UI ---
-with gr.Blocks(title="Email Gatekeeper AI") as demo:
+# Gradio Setup
+with gr.Blocks() as demo:
     gr.Markdown("# 📧 Email Gatekeeper AI")
-    with gr.Row():
-        diff = gr.Dropdown(choices=["easy", "medium", "hard"], value="easy", label="Select Difficulty")
-        btn = gr.Button("Analyze Emails", variant="primary")
-    out = gr.Textbox(label="Evaluation Logs", lines=12)
+    diff = gr.Dropdown(["easy", "medium", "hard"], value="easy", label="Difficulty")
+    btn = gr.Button("Analyze Emails")
+    out = gr.Textbox(label="Logs", lines=10)
     btn.click(run_demo, inputs=diff, outputs=out)
 
-# IS LINE KO DHAYAN SE DEKHO: Humne path="/" kar diya hai
 app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":
-    # Hugging Face hamesha port 7860 use karta hai
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    if len(sys.argv) > 1 and sys.argv[1] == "--test":
+        main()
+    else:
+        uvicorn.run(app, host="0.0.0.0", port=7860)