Upload 14 files

Dockerfile

@@ -0,0 +1,3 @@
+# Minimal Dockerfile for OpenEnv 
+
+ENV ENABLE_WEB_INTERFACE=true

README.md

@@ -1,10 +1,9 @@
----
-title: SmartEmailAI
-emoji: 🦀
-colorFrom: gray
-colorTo: purple
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+sdk: gradio
+app_file: client.py
+colorFrom: blue
+colorTo: green
+base_path: /web
+---
+# Email Env Project
+Minimal OpenEnv/HF testing environment

agent.py

@@ -0,0 +1,13 @@
+from transformers import pipeline
+
+classifier = pipeline("text-classification")
+
+def llm_agent(observation):
+    result = classifier(observation)[0]['label'].lower()
+
+    if "spam" in result:
+        return "spam"
+    elif "important" in result:
+        return "important"
+    else:
+        return "normal"

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from fastapi import FastAPI
+from env import EmailEnv
+from agent import llm_agent
+from rl import update_q
+
+app = FastAPI()
+
+env = EmailEnv()
+obs = None
+total_score = 0
+
+@app.post("/reset")
+def reset_env():
+    global obs, total_score
+    obs = env.reset()
+    total_score = 0
+    return {"status": "success"}
+
+@app.get("/")
+def home():
+    return {"message": "running"}
+
+def step_env(user_action):
+    global obs, total_score
+    if obs is None:
+        obs = env.reset()
+        total_score = 0
+        return obs, 0, total_score
+    obs_new, reward, done, info = env.step(user_action)
+    total_score += reward
+    if done:
+        obs = None
+        return "Finished Click again to restart", reward, total_score
+    return obs_new, reward, total_score
+
+def auto_ai():
+    global obs, total_score
+    if obs is None:
+        obs = env.reset()
+        total_score = 0
+        return obs, 0, total_score
+    action = llm_agent(obs)
+    obs_new, reward, done, info = env.step(action)
+    total_score += reward
+    if done:
+        obs = None
+        return "AI Finished Click again to restart", reward, total_score
+    return f"AI chose: {action} | {obs_new}", reward, total_score
+
+iface = gr.Interface(
+    fn=step_env,
+    inputs=gr.Radio(["important", "spam", "normal"]),
+    outputs=["text", "number", "number"],
+    title="Smart Email AI Trainer"
+)
+
+app = gr.mount_gradio_app(app, iface, path="/")

client.py

@@ -0,0 +1,114 @@
+from pydantic import BaseModel
+from typing import List, Literal
+import gradio as gr
+
+class EmailAction(BaseModel):
+    action_type: Literal['send', 'reply', 'archive', 'delete']
+    message: str = ""
+    email_id: int = None
+
+class EmailObservation(BaseModel):
+    emails: List[dict]
+    ai_feedback: str
+
+class EmailEnv:
+    def __init__(self):
+        self.emails = [
+            {"id": 1, "from": "alice@example.com", "subject": "Meeting Tomorrow", "label": "Work", "status": "Unread"},
+            {"id": 2, "from": "bob@example.com", "subject": "Lunch Plans", "label": "Personal", "status": "Archived"},
+            {"id": 3, "from": "carol@example.com", "subject": "Project Update", "label": "Work", "status": "Sent"},
+            {"id": 4, "from": "dave@spam.com", "subject": "Win a Prize", "label": "Spam", "status": "Deleted"},
+        ]
+
+    def step(self, action: EmailAction) -> EmailObservation:
+        feedback = ""
+        target = next((e for e in self.emails if e["id"] == action.email_id), None) if action.email_id else None
+        if action.action_type == "send":
+            new_id = max([e["id"] for e in self.emails]+[0]) + 1
+            self.emails.append({
+                "id": new_id,
+                "from": "me@example.com",
+                "subject": "Project Update",
+                "label": "Work",
+                "status": "Sent",
+                "body": action.message
+            })
+            feedback = "Action Executed: Send Email ✅"
+        elif action.action_type == "reply" and target:
+            new_id = max([e["id"] for e in self.emails]+[0]) + 1
+            self.emails.append({
+                "id": new_id,
+                "from": "me@example.com",
+                "subject": f"Re: {target['subject']}",
+                "label": target["label"],
+                "status": "Sent",
+                "body": action.message
+            })
+            feedback = f"Action Executed: Reply ✅ to email ID {action.email_id}"
+        elif action.action_type == "archive" and target:
+            target["status"] = "Archived"
+            feedback = f"Action Executed: Archive ✅ email ID {action.email_id}"
+        elif action.action_type == "delete" and target:
+            target["status"] = "Deleted"
+            feedback = f"Action Executed: Delete ✅ email ID {action.email_id}"
+        else:
+            feedback = "Invalid action or email ID."
+        return EmailObservation(emails=self.emails, ai_feedback=feedback)
+
+env = EmailEnv()
+
+status_colors = {
+    "Unread": "orange",
+    "Sent": "green",
+    "Archived": "blue",
+    "Deleted": "red"
+}
+
+def render_table_html(emails):
+    table_html = "<table style='border-collapse: collapse; width: 100%;'>"
+    table_html += "<tr><th>ID</th><th>From</th><th>Subject</th><th>Label</th><th>Status</th></tr>"
+    for e in emails:
+        color = status_colors.get(e["status"], "white")
+        table_html += f"<tr style='border:1px solid #ddd;'>"
+        table_html += f"<td>{e['id']}</td>"
+        table_html += f"<td>{e['from']}</td>"
+        table_html += f"<td>{e['subject']}</td>"
+        table_html += f"<td>{e['label']}</td>"
+        table_html += f"<td style='background-color:{color}; color:white;'>{e['status']}</td>"
+        table_html += "</tr>"
+    table_html += "</table>"
+    return table_html
+
+def gradio_step(action_type, message, email_id):
+    if not action_type:
+        return "Invalid action selected.", ""
+    action_type_safe = str(action_type).lower()
+    email_id_val = None
+    if action_type_safe != "send" and email_id:
+        email_id_str = str(email_id).strip()
+        if email_id_str != "":
+            try:
+                email_id_val = int(email_id_str)
+            except ValueError:
+                return "Invalid Email ID. Must be a number.", ""
+    obs = env.step(EmailAction(action_type=action_type_safe, message=message, email_id=email_id_val))
+    table_html = render_table_html(obs.emails)
+    return obs.ai_feedback, table_html
+
+iface = gr.Interface(
+    fn=gradio_step,
+    inputs=[
+        gr.Dropdown(['Send', 'Reply', 'Archive', 'Delete'], label="Choose Action"),
+        gr.Textbox(label="Message"),
+        gr.Textbox(label="Email ID (for reply/archive/delete)")
+    ],
+    outputs=[
+        gr.Textbox(label="AI Feedback"),
+        gr.HTML(label="Emails Dashboard")
+    ],
+    live=False,
+    title="Smart Email Management AI Environment (OpenEnv)",
+    description="Simulated email environment with AI-powered actions: send, reply, archive, delete."
+)
+
+iface.launch(server_name="0.0.0.0", server_port=7860)

env.py

@@ -0,0 +1,27 @@
+class EmailEnv:
+    def __init__(self):
+        self.data = []
+        self.index = 0
+
+    def reset(self):
+        self.data = [
+            {"text": "Meeting at 5pm", "label": "important"},
+            {"text": "Win a free iPhone", "label": "spam"},
+            {"text": "Weekly report attached", "label": "normal"}
+        ]
+        self.index = 0
+        return self.data[self.index]["text"]
+
+    def step(self, action):
+        if self.index >= len(self.data):
+            return None, 0, True, {}
+
+        correct = self.data[self.index]["label"]
+        reward = 10 if action == correct else -5
+
+        self.index += 1
+        done = self.index >= len(self.data)
+
+        next_obs = None if done else self.data[self.index]["text"]
+
+        return next_obs, reward, done, {"correct": correct}

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

inference.py

@@ -0,0 +1,48 @@
+from typing import List, Literal
+from pydantic import BaseModel
+
+class EmailAction(BaseModel):
+    action_type: Literal['send', 'reply', 'archive', 'delete']
+    message: str = ""
+    email_id: int = None
+
+class EmailEnv:
+    def __init__(self):
+        self.reset()
+
+    def reset(self):
+        self.emails = [
+            {"id": 1, "from": "alice@example.com", "subject": "Meeting Tomorrow", "label": "Work", "status": "Unread"},
+            {"id": 2, "from": "bob@example.com", "subject": "Lunch Plans", "label": "Personal", "status": "Archived"},
+        ]
+        return {"emails": self.emails}
+
+    def step(self, action: dict):
+        if action["action_type"] == "send":
+            new_id = len(self.emails) + 1
+            self.emails.append({
+                "id": new_id,
+                "from": "me@example.com",
+                "subject": "New Mail",
+                "label": "Work",
+                "status": "Sent"
+            })
+            return {"emails": self.emails, "reward": 1, "done": False}
+        return {"emails": self.emails, "reward": 0, "done": False}
+
+env = EmailEnv()
+obs = None
+
+def reset():
+    global obs
+    obs = env.reset()
+    return obs
+
+def step(action):
+    global obs
+    result = env.step(action)
+    obs = result["emails"]
+    return result
+
+def act():
+    return {"action_type": "send", "message": "Hello"}

openenv.yaml

@@ -0,0 +1,3 @@
+name: email_env
+description: Minimal OpenEnv environment
+version: 0.1.0

pyproject.toml

@@ -0,0 +1,3 @@
+[build-system] 
+requires = ["setuptools", "wheel"] 
+build-backend = "setuptools.build_meta" 

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+pydantic
+transformers
+torch

rl.py

@@ -0,0 +1,12 @@
+Q = {}
+actions = ["important", "spam", "normal"]
+
+def get_action(state):
+    if state not in Q:
+        Q[state] = {a: 0 for a in actions}
+    return max(Q[state], key=Q[state].get)
+
+def update_q(state, action, reward):
+    if state not in Q:
+        Q[state] = {a: 0 for a in actions}
+    Q[state][action] += reward