fixed so that neon connection doesn't randomly die

Author: codinglabsong

api/llm/agent.py

@@ -14,19 +14,35 @@
 
 # Global agent instance
 _agent_executor = None
+_checkpointer = None
 
 
 @lru_cache
 def get_checkpointer():
-    DATABASE_URL = os.environ.get("DATABASE_URL")
-    if not DATABASE_URL:
-        raise RuntimeError("DATABASE_URL is not set. Point it to your Neon connection string.")
-
-    cm = PostgresSaver.from_conn_string(DATABASE_URL)
-    saver = cm.__enter__()  # enter the context manager once
-    atexit.register(lambda: cm.__exit__(None, None, None))  # clean shutdown
-    saver.setup()  # create tables on first run; no-op afterward
-    return saver
+    """Open PostgresSaver once and reuse it (with keepalives)."""
+    global _checkpointer
+
+    if _checkpointer is None:
+        url = os.environ.get("DATABASE_URL")
+        if not url:
+            raise RuntimeError("DATABASE_URL is not set. Point it to your Neon connection string.")
+
+        # add keepalive params if missing
+        if "keepalives=" not in url:
+            sep = "&" if "?" in url else "?"
+            url += (
+                sep
+                + "sslmode=require&keepalives=1&keepalives_idle=30&keepalives_interval=10\
+                    &keepalives_count=3"
+            )
+
+        cm = PostgresSaver.from_conn_string(url)
+        saver = cm.__enter__()  # enter the context manager once
+        atexit.register(lambda: cm.__exit__(None, None, None))  # clean shutdown
+        saver.setup()  # create tables on first run; no-op afterward
+        _checkpointer = saver
+
+    return _checkpointer
 
 
 def get_agent():

src/app/page.tsx

@@ -22,6 +22,7 @@ const SAMPLE_IMAGES: ImageItem[] = [
     description:
       "A beautiful landscape with mountains and a serene lake reflecting the sky.",
     timestamp: new Date(Date.now() - 86400000), // 1 day ago
+    type: "sample",
   },
   {
     id: "img_002",
@@ -30,6 +31,7 @@ const SAMPLE_IMAGES: ImageItem[] = [
     description:
       "Urban cityscape with modern architecture and vibrant street life.",
     timestamp: new Date(Date.now() - 43200000), // 12 hours ago
+    type: "sample",
   },
   {
     id: "img_003",
@@ -38,26 +40,27 @@ const SAMPLE_IMAGES: ImageItem[] = [
     description:
       "Portrait photography with dramatic lighting and artistic composition.",
     timestamp: new Date(),
+    type: "sample",
   },
 ];
 
 export default function Home() {
   const [messages, setMessages] = useState<Message[]>([INITIAL_MESSAGE]);
   const [selectedImages, setSelectedImages] = useState<Set<string>>(new Set());
-  const [images] = useState<ImageItem[]>(SAMPLE_IMAGES);
+  const [images, setImages] = useState<ImageItem[]>(SAMPLE_IMAGES);
   const [isLoading, setIsLoading] = useState(false);
   const [userId] = useState(() => crypto.randomUUID());
 
-  // Scroll to the right end (newest images) on mount and window resize
-  useEffect(() => {
+  // Reusable scroll function
+  const scrollToRight = () => {
     const scrollContainer = document.getElementById("image-scroll-container");
+    if (scrollContainer) {
+      scrollContainer.scrollLeft = scrollContainer.scrollWidth;
+    }
+  };
 
-    const scrollToRight = () => {
-      if (scrollContainer) {
-        scrollContainer.scrollLeft = scrollContainer.scrollWidth;
-      }
-    };
-
+  // Scroll to the right end (newest images) on mount and window resize
+  useEffect(() => {
     // Initial scroll
     setTimeout(scrollToRight, 1);
 
@@ -133,6 +136,22 @@ export default function Home() {
   };
 
   const handleChatImageUpload = async (file: File) => {
+    // Create uploaded image item with UUID
+    const uploadedImage: ImageItem = {
+      id: crypto.randomUUID(),
+      url: URL.createObjectURL(file),
+      title: "",
+      description: "",
+      timestamp: new Date(),
+      type: "uploaded",
+    };
+
+    // Add to images list
+    setImages((prev) => [...prev, uploadedImage]);
+
+    // Scroll to show the new image
+    setTimeout(scrollToRight, 100);
+
     const uploadMessage = `📷 Uploaded image: ${file.name}`;
 
     // Add user message

src/lib/types.ts

@@ -11,4 +11,5 @@ export interface ImageItem {
   title: string;
   description: string;
   timestamp: Date;
+  type: "uploaded" | "generated" | "sample";
 }