Initial version with just scenarios to push into sxpaces

.env.example

@@ -0,0 +1 @@
+OPENAI_API_KEY='key'

.gitignore

@@ -0,0 +1,20 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# System junk
+.DS_Store
+
+# Virtual environments
+.venv
+.env
+venv/
+
+# Generated images
+generated_images/
+images/
+*.png

.python-version

@@ -0,0 +1 @@
+3.11

app/app.py

@@ -0,0 +1,127 @@
+import gradio as gr
+import json
+from pathlib import Path
+from openai import OpenAI
+import os
+from dotenv import load_dotenv
+import base64
+import requests
+import time
+
+
+# Load environment variables from .env file
+load_dotenv()
+
+OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
+
+client = OpenAI(api_key=OPENAI_API_KEY)
+models = client.models.list()
+
+
+TEMPLATES_DIR = Path(__file__).resolve().parent.parent / "mcp_server" / "templates"
+GENERATED_IMAGES_DIR = Path(__file__).resolve().parent.parent / "generated_images"
+GENERATED_IMAGES_DIR.mkdir(exist_ok=True)
+
+
+def load_scenarios():
+    return [p.stem for p in TEMPLATES_DIR.glob("*.json")]
+    
+
+def generate_story(scenario_name: str):
+    filepath = TEMPLATES_DIR / f"{scenario_name}.json"
+    if not filepath.exists():
+        return "Template not found."
+    template = json.loads(filepath.read_text())
+    story_prompt = (
+        "Rewrite the following scenario into a simple, autism-friendly social story.\n"
+        "Use short sentences, gentle tone, clear structure, and supportive language.\n"
+        "Do NOT return JSON. Return only the story text.\n\n"
+        f"Scenario data:\n{json.dumps(template, indent=2)}"
+    )
+    story_resp = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[{"role": "user", "content": story_prompt}]
+    )
+    return story_resp.choices[0].message.content.strip()
+
+
+def generate_image(scenario_name: str):
+    filepath = TEMPLATES_DIR / f"{scenario_name}.json"
+    if not filepath.exists():
+        return None
+    template = json.loads(filepath.read_text())
+    image_prompt = (
+        f"Create a soft, calming illustration for a social story about: {template['title']}."
+        " Use pastel colors, simple shapes, friendly characters, minimal details, no text."
+    )
+    # Comment out image generation while developing
+    # img = client.images.generate(
+    #     model="gpt-image-1",
+    #     prompt=image_prompt,
+    #     output_format="png",
+    #     size="512x512"
+    # )
+    # image_bytes = base64.b64decode(img.data[0].b64_json)
+    # timestamp = int(time.time())
+    # image_filename = f"{scenario_name}_{timestamp}.png"
+    # image_path = GENERATED_IMAGES_DIR / image_filename
+    # with open(image_path, "wb") as f:
+    #     f.write(image_bytes)
+    # img_src = str(image_path)
+    # return img_src
+    # For now, return None or a public image URL
+    return None
+
+
+# def main():
+#     print("Loading scenarios...")
+#     scenarios = load_scenarios()
+#     print("Scenarios loaded:", scenarios)
+#
+#     with gr.Blocks() as demo:
+#         gr.Markdown("# 🧸 Comfortool\n### Autism-friendly social stories with calming illustrations")
+#
+#         dropdown = gr.Dropdown(
+#             choices=scenarios,
+#             label="Choose a scenario"
+#         )
+#
+#         generate_btn = gr.Button("Generate Social Story")
+#         story_out = gr.Textbox(label="Story", lines=12)
+#         image_out = gr.Image(label="Illustration")
+#
+#         def on_generate(scenario_name):
+#             print("Generating story for:", scenario_name)
+#             story = generate_story(scenario_name)
+#             print("Story generated:", story)
+#             image = None  # For now, do not generate image
+#             return story, image
+#
+#         generate_btn.click(
+#             fn=on_generate,
+#             inputs=dropdown,
+#             outputs=[story_out, image_out]
+#         )
+#
+#     print("Gradio app initialized.")
+#     return demo
+
+
+def show_selected(scenario_name):
+    return f"You selected: {scenario_name}"
+
+def main():
+    scenarios = load_scenarios()
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🧸 Comfortool\n### Available scenarios")
+        dropdown = gr.Dropdown(choices=scenarios, label="Choose a scenario")
+        output = gr.Textbox(label="Selected scenario")
+        dropdown.change(fn=show_selected, inputs=dropdown, outputs=output)
+    return demo
+
+
+if __name__ == "__main__":
+    demo = main()
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+
+

mcp_server/server.py

@@ -0,0 +1,39 @@
+from mcp.server.fastmcp import FastMCP
+from pathlib import Path
+import json
+
+# Initialize MCP server
+mcp = FastMCP("comfortool-mcp-server")
+
+BASE_DIR = Path(__file__).resolve().parent
+TEMPLATE_DIR = BASE_DIR / "templates"
+IMAGES_DIR = BASE_DIR / "images"
+
+
+# ---- Tool: Get list of scenarios ----
+@mcp.tool()
+def get_scenarios() -> list[str]:
+    """Return the list of available scenario template names."""
+    return [p.stem for p in TEMPLATE_DIR.glob("*.json")]
+
+
+# ---- Tool: Get a specific scenario template ----
+@mcp.tool()
+def get_template(name: str) -> dict:
+    """Load and return a scenario JSON template by name."""
+    filepath = TEMPLATE_DIR / f"{name}.json"
+    if not filepath.exists():
+        raise FileNotFoundError(f"Template '{name}' not found.")
+    return json.loads(filepath.read_text())
+
+
+# ---- Tool: List icons/images (optional for now) ----
+@mcp.tool()
+def get_static_images() -> list[str]:
+    """Return the filenames of static images/icons."""
+    return [p.name for p in IMAGES_DIR.glob("*")]
+
+
+# ---- Run the server ----
+if __name__ == "__main__":
+    mcp.run()

mcp_server/templates/dentist.json

@@ -0,0 +1,8 @@
+{
+  "id": "going_to_the_dentist",
+  "title": "Going to the Dentist",
+  "story": "Today I am going to the dentist. The dentist helps keep my teeth clean and healthy. First, I will enter the quiet waiting room. Then, someone will call my name. I will sit in a big chair that goes up and down. The dentist will look at my teeth and might clean them. Some sounds may be loud, but they are safe. If I need a break, I can ask for one. When it is finished, I can go home feeling proud.",
+  "sensory": ["bright light", "whirring sounds"],
+  "tools": ["headphones", "fidget", "sunglasses"],
+  "emotion_support": "It is okay to feel nervous. Many people do."
+}

mcp_server/templates/doctor.json

@@ -0,0 +1,5 @@
+{
+  "id": "going_to_the_doctor",
+  "title": "Going to the Doctor",
+  "story": "Today I am visiting the doctor. The doctor keeps my body healthy. I will check in and wait until my name is called. The doctor may ask questions and gently check parts of my body. Sometimes, the doctor might use tools like a stethoscope. If I need a moment, I can say 'please wait.' After the visit, I can go home knowing I took good care of myself."
+}

mcp_server/templates/first_day_school.json

@@ -0,0 +1,5 @@
+{
+  "id": "first_day_of_school",
+  "title": "First Day of School",
+  "story": "Today is my first day of school. I will meet teachers and classmates. I will learn where my classroom is and where I can take breaks. There will be new routines, and that is okay. If things feel confusing, I can ask a teacher for help. Each day will feel more familiar."
+}

mcp_server/templates/haircut.json

@@ -0,0 +1,5 @@
+{
+  "id": "getting_a_haircut",
+  "title": "Getting a Haircut",
+  "story": "Today I am getting a haircut. The hairdresser helps my hair look nice. First, I will sit in a chair. Someone may touch my hair or use scissors. Scissors make small sounds, but they are safe. Hair may fall on my skin and feel tickly, but it can be brushed away. After the haircut, my head will feel lighter and fresh."
+}

mcp_server/templates/loud_events.json

@@ -0,0 +1,5 @@
+{
+  "id": "loud_and_crowded_events",
+  "title": "Loud and Crowded Places",
+  "story": "Today I might be in a loud or busy place. There may be many people and sounds. If it feels too loud, I can use headphones or take a break. I can stay close to someone I trust. I do not have to talk if I don’t want to. When the event is over, I can rest somewhere quiet."
+}

mcp_server/templates/new_food.json

@@ -0,0 +1,5 @@
+{
+  "id": "trying_a_new_food",
+  "title": "Trying a New Food",
+  "story": "Today I might try a new food. New food has a new smell, texture, or taste. I can look at the food first, smell it, touch it, or lick it. If I don’t like it, that is okay. Trying is already a success."
+}

mcp_server/templates/new_place.json

@@ -0,0 +1,5 @@
+{
+  "id": "visiting_a_new_place",
+  "title": "Visiting a New Place",
+  "story": "Today I will go somewhere I have not been before. New places can feel exciting or scary. I can look around slowly and learn what is there. I can stay close to someone I trust. When I’m ready, I can explore more. Every new place becomes familiar after a while."
+}

mcp_server/templates/physical_education.json

@@ -0,0 +1,5 @@
+{
+  "id": "physical_education_class",
+  "title": "PE Class",
+  "story": "In PE class, I will move my body and try fun activities. There may be loud sounds like balls bouncing or whistles. I can follow instructions at my own pace. If an activity feels too fast or confusing, I can ask for a pause. PE is for having fun and learning new skills."
+}

mcp_server/templates/pickup_from_school.json

@@ -0,0 +1,5 @@
+{
+  "id": "getting_picked_up",
+  "title": "Getting Picked Up From School",
+  "story": "When school is over, someone comes to pick me up. If I am playing, stopping can feel hard. I will get a reminder before it’s time to go. I can take one last turn or finish my activity. When I’m ready, I will leave calmly and go home."
+}

mcp_server/templates/social_situations.json

@@ -0,0 +1,5 @@
+{
+  "id": "social_situations",
+  "title": "Social Situations",
+  "story": "Sometimes I will be around people and talk to them. I don’t have to talk a lot. I can use short words or gestures. If I need space, I can step away for a moment. Everyone communicates in different ways, and that is okay. Being myself is enough."
+}

pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "comfortool"
+version = "0.0.1"
+requires-python = ">=3.10"
+description = "Comfortool - autism-friendly social stories"
+dependencies = [
+    "fastapi>=0.122.0",
+    "gradio>=6.0.1",
+    "openai>=2.8.1",
+    "pillow>=12.0.0",
+    "python-dotenv>=1.2.1",
+    "requests>=2.32.5",
+    "uvicorn>=0.38.0",
+]
+
+[tool.uv]
+# uv config can go here if needed (optional)

requirements.txt

@@ -0,0 +1,4 @@
+openai
+gradio
+python-dotenv
+requests