Omkar1806 commited on
Commit
c28bcbd
·
verified ·
1 Parent(s): 8c0b2af

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -66
app.py DELETED
@@ -1,66 +0,0 @@
1
- import sys
2
- import os
3
- import uvicorn
4
- import numpy as np
5
- import gradio as gr
6
- from fastapi import FastAPI
7
- from env import EmailTriageEnv, URGENCY_LABELS, ROUTING_LABELS, RESOLUTION_LABELS
8
-
9
- app = FastAPI()
10
-
11
- # --- Agent Logic ---
12
- def smart_agent_logic(desc):
13
- desc = desc.lower()
14
- if any(x in desc for x in ["password", "hacked", "breach", "phish", "threat", "ransomware"]):
15
- return [2, 2, 2] if "threat" in desc or "ransomware" in desc else [2, 1, 2]
16
- if any(x in desc for x in ["billing", "refund", "dispute", "invoice", "payment"]):
17
- return [1, 2, 2]
18
- if any(x in desc for x in ["support", "routine", "slow", "error"]):
19
- return [0, 1, 1]
20
- return [0, 0, 0]
21
-
22
- # --- Core Function ---
23
- def run_demo(task):
24
- try:
25
- env = EmailTriageEnv(task=task)
26
- env.reset()
27
- results = []
28
- total_reward = 0
29
- print(f"[START] Task: {task}")
30
- for i, email in enumerate(env._queue):
31
- action = smart_agent_logic(email['description'])
32
- _, reward, _, _, _ = env.step(action)
33
- total_reward += reward
34
- print(f"[STEP] Index: {i} | Action: {action} | Reward: {reward}")
35
- status = "✅ MATCH" if reward >= 1.0 else "❌ MISMATCH"
36
- results.append(f"#{i+1} [{task.upper()}] {email['description'][:30]}... | {status}")
37
- score = total_reward / len(env._queue) if env._queue else 0
38
- print(f"[END] Final Score: {score}")
39
- return "\n".join(results) + f"\n\n--- FINAL SCORE: {score:.3f} / 1.000 ---"
40
- except Exception as e:
41
- return f"Error: {str(e)}"
42
-
43
- # --- API for Validator ---
44
- @app.post("/reset")
45
- async def reset_endpoint():
46
- return {"status": "success", "message": "Environment Reset OK"}
47
-
48
- @app.get("/status")
49
- async def health_check():
50
- return {"status": "online"}
51
-
52
- # --- Gradio UI ---
53
- with gr.Blocks(title="Email Gatekeeper AI") as demo:
54
- gr.Markdown("# 📧 Email Gatekeeper AI")
55
- with gr.Row():
56
- diff = gr.Dropdown(choices=["easy", "medium", "hard"], value="easy", label="Select Difficulty")
57
- btn = gr.Button("Analyze Emails", variant="primary")
58
- out = gr.Textbox(label="Evaluation Logs", lines=12)
59
- btn.click(run_demo, inputs=diff, outputs=out)
60
-
61
- # IS LINE KO DHAYAN SE DEKHO: Humne path="/" kar diya hai
62
- app = gr.mount_gradio_app(app, demo, path="/")
63
-
64
- if __name__ == "__main__":
65
- # Hugging Face hamesha port 7860 use karta hai
66
- uvicorn.run(app, host="0.0.0.0", port=7860)