Adds export functionality to chat, removes poetry in favor of uv, bumps Honcho version to v0.0.14/python-SDK-0.0.19

Dockerfile

@@ -2,30 +2,24 @@
 # https://testdriven.io/blog/docker-best-practices/
 FROM python:3.11-slim-bullseye
 
-RUN apt-get update && apt-get install -y build-essential
+RUN apt-get update && apt-get install -y build-essential curl
 
 WORKDIR /app
 
-# https://stackoverflow.com/questions/53835198/integrating-python-poetry-with-docker
+# Set Python environment variables and default port
 ENV PYTHONFAULTHANDLER=1 \
-  PYTHONUNBUFFERED=1 \
-  PYTHONHASHSEED=random \
-  PIP_NO_CACHE_DIR=off \
-  PIP_DISABLE_PIP_VERSION_CHECK=on \
-  PIP_DEFAULT_TIMEOUT=100 \
-  POETRY_VERSION=1.8.3
+    PYTHONUNBUFFERED=1 \
+    PYTHONHASHSEED=random \
+    PORT=8000
 
-RUN pip install "poetry==$POETRY_VERSION"
+# Install uv and add to PATH
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
+    mv /root/.local/bin/uv /usr/local/bin/ && \
+    mv /root/.local/bin/uvx /usr/local/bin/
 
-# Copy only requirements to cache them in docker layer
-WORKDIR /app
-COPY poetry.lock pyproject.toml /app/
-
-# Project initialization:
-RUN poetry config virtualenvs.create false \
-  && poetry install --no-root --no-interaction --no-ansi
-
-WORKDIR /app
+# Copy requirements and install dependencies
+COPY requirements.txt .
+RUN uv pip install --system -r requirements.txt
 
 RUN addgroup --system app && adduser --system --group app
 RUN chown -R app:app /app
@@ -34,7 +28,5 @@ USER app
 COPY --chown=app:app app.py /app/app.py
 COPY --chown=app:app calls.py /app/calls.py
 
-EXPOSE 8000
-
-# https://stackoverflow.com/questions/29663459/python-app-does-not-print-anything-when-running-detached-in-docker
-CMD ["fastapi", "run", "app.py", "--host", "0.0.0.0", "--port", "8000"]
+EXPOSE ${PORT}
+CMD uvicorn app:app --host 0.0.0.0 --port ${PORT}

app.py

@@ -1,26 +1,30 @@
 import os
 from contextvars import ContextVar
-from typing import Annotated, Any, Dict
+from typing import Annotated, Any, Dict, List, Literal
 from functools import cache
 from fastapi.security import OAuth2PasswordBearer
+from fastapi.responses import HTMLResponse, FileResponse
+from fastapi.staticfiles import StaticFiles
 
 import sentry_sdk
 from fastapi import Depends, FastAPI, HTTPException, status
 from starlette.middleware.cors import CORSMiddleware
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, FileResponse
 from honcho import Honcho
 from pydantic import BaseModel
 
 from cryptography.fernet import Fernet
 import base64
 
 from calls import GaslitClaude, Simulator
-
 import jwt
 
-
 from dotenv import load_dotenv
 
+import tempfile
+import json
+from datetime import datetime
+
 load_dotenv(override=True)
 
 
@@ -35,13 +39,23 @@ def get_env(key: str):
 CLIENT_REGEX = get_env("CLIENT_REGEX")
 print(CLIENT_REGEX)
 JWT_SECRET = get_env("JWT_SECRET")
-SECRET_KEY = base64.b64decode(get_env("SECRET_KEY"))
+SECRET_KEY = get_env("SECRET_KEY").encode()
 HONCHO_APP_NAME = get_env("HONCHO_APP_NAME")
 
 fernet = Fernet(SECRET_KEY)
 
-honcho = Honcho(base_url=HONCHO_ENV)
-honcho_app = honcho.apps.get_or_create(HONCHO_APP_NAME)
+print(f"Initializing Honcho with base_url: {HONCHO_ENV}")
+honcho = Honcho(
+    base_url=HONCHO_ENV,
+)
+
+try:
+    print(f"Attempting to get/create app: {HONCHO_APP_NAME}")
+    honcho_app = honcho.apps.get_or_create(HONCHO_APP_NAME)
+    print(f"Successfully initialized app with id: {honcho_app.id}")
+except Exception as e:
+    print(f"Error initializing Honcho app: {str(e)}")
+    raise
 
 
 gaslit_ctx = ContextVar(
@@ -72,7 +86,6 @@ def get_env(key: str):
     allow_headers=["*"],
 )
 
-
 class BaseRequest(BaseModel):
     session_id: str
 
@@ -212,10 +225,24 @@ async def reset(
         honcho.apps.users.sessions.delete(
             app_id=honcho_app.id, session_id=session_id, user_id=user_id
         )
-    session = honcho.apps.users.sessions.create(app_id=honcho_app.id, user_id=user_id)
     # TODO reset the session
     # gaslit_claude.history = []
     # simulator.history = []
+    try:
+        session = honcho.apps.users.sessions.create(
+            app_id=honcho_app.id, 
+            user_id=user_id,
+        )
+    except TypeError as e:
+        if "location_id" in str(e):
+            # If location_id is truly optional, try without it
+            session = honcho.apps.users.sessions.create(
+                app_id=honcho_app.id,
+                user_id=user_id
+            )
+        else:
+            raise e
+
     return {
         "user_id": user_id,
         "session_id": session.id,
@@ -299,3 +326,44 @@ async def share_messages(code: str):
         return await get_session_messages(session_id=session_id, user_id=user_id)
     except Exception:
         raise HTTPException(status_code=400, detail="Invalid encrypted data")
+
+class MessageFormat(BaseModel):
+    role: Literal["user", "assistant"]
+    content: str
+
+@app.get("/export/{session_id}")
+async def export_session(session_id: str, user_id: str = Depends(get_current_user)):
+    try:
+        messages = honcho.apps.users.sessions.messages.list(
+            app_id=honcho_app.id,
+            user_id=user_id, 
+            session_id=session_id
+        )
+
+        formatted_messages = [
+            {
+                "role": "user" if msg.is_user else "assistant",
+                "content": msg.content
+            } for msg in messages
+        ]
+
+        # Create a temporary file
+        with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.json') as tmp:
+            json.dump(formatted_messages, tmp, indent=2)
+            tmp_path = tmp.name
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"yousim_conversation_{timestamp}.json"
+
+        return FileResponse(
+            path=tmp_path,
+            filename=filename,
+            media_type='application/json',
+            background=None  # Ensures file is sent before deletion
+        )
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Failed to export session: {str(e)}"
+        )

docker-compose.yml

@@ -1,12 +1,15 @@
-version: '3.8'
+# version: '3.8'
 
 services:
   backend:
     build:
       context: .
       dockerfile: Dockerfile
     ports:
-      - "8000:8000"
+      - "${PORT}:${PORT}"
+    environment:
+      - HONCHO_AUTH_TOKEN=default
+      - PORT=${PORT}
     env_file:
       - .env
 
@@ -18,3 +21,5 @@ services:
       - "3000:3000"
     depends_on:
       - backend
+    env_file:
+      - ./webshell/.env