Update app.py

app.py

@@ -8,6 +8,7 @@ import zipfile
 import re
 import logging
 import asyncio
+import random  # Added for ModelBuilder jokes
 from io import BytesIO
 from datetime import datetime
 import pytz
@@ -19,6 +20,7 @@ import pandas as pd
 import torch
 import fitz
 import requests
+import aiofiles  # Added for async file operations
 from PIL import Image
 from diffusers import StableDiffusionPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
@@ -272,7 +274,7 @@ def process_image_with_prompt(image, prompt, model="o3-mini-high"):
     except Exception as e:
         return f"Error processing image with GPT: {str(e)}"
 
-# --- Sidebar Setup (Add this near the top, before tabs) ---
+# --- Sidebar Setup ---
 st.sidebar.subheader("Gallery Settings")
 if 'gallery_size' not in st.session_state:
     st.session_state['gallery_size'] = 2  # Default value
@@ -288,7 +290,6 @@ def update_gallery():
     if all_files:
         st.sidebar.subheader("Asset Gallery 📸📖")
         cols = st.sidebar.columns(2)
-        # Use the stored gallery size from session state
         for idx, file in enumerate(all_files[:st.session_state['gallery_size']]):
             with cols[idx % 2]:
                 st.session_state['unique_counter'] += 1
@@ -315,10 +316,9 @@ def update_gallery():
                     st.sidebar.success(f"Asset {os.path.basename(file)} vaporized! 💨")
                     st.rerun()
 
-# Call update_gallery() once initially or after actions as needed
+# Call update_gallery() once initially
 update_gallery()
 
-
 # --- Sidebar Logs & History ---
 st.sidebar.subheader("Action Logs 📜")
 with st.sidebar:
@@ -329,7 +329,7 @@ with st.sidebar:
     for entry in st.session_state['history']:
         st.write(entry)
 
-# --- Create Tabs (Existing + New) ---
+# --- Create Tabs ---
 tabs = st.tabs([
     "Camera Snap 📷",
     "Download PDFs 📥",
@@ -342,7 +342,7 @@ tabs = st.tabs([
 ])
 (tab_camera, tab_download, tab_ocr, tab_build, tab_imggen, tab_pdf_process, tab_image_process, tab_md_gallery) = tabs
 
-# === Tab: Camera Snap (existing) ===
+# === Tab: Camera Snap ===
 with tab_camera:
     st.header("Camera Snap 📷")
     st.subheader("Single Capture")
@@ -378,7 +378,7 @@ with tab_camera:
             logger.info(f"Saved snapshot from Camera 1: {filename}")
             update_gallery()
 
-# === Tab: Download PDFs (existing) ===
+# === Tab: Download PDFs ===
 with tab_download:
     st.header("Download PDFs 📥")
     if st.button("Examples 📚"):
@@ -439,7 +439,7 @@ with tab_download:
         else:
             st.warning("No PDFs selected for snapshotting! Check some boxes in the sidebar.")
 
-# === Tab: Test OCR (existing) ===
+# === Tab: Test OCR ===
 with tab_ocr:
     st.header("Test OCR 🔍")
     all_files = get_gallery_files()
@@ -505,7 +505,7 @@ with tab_ocr:
     else:
         st.warning("No assets in gallery yet. Use Camera Snap or Download PDFs!")
 
-# === Tab: Build Titan (existing) ===
+# === Tab: Build Titan ===
 with tab_build:
     st.header("Build Titan 🌱")
     model_type = st.selectbox("Model Type", ["Causal LM", "Diffusion"], key="build_type")
@@ -529,7 +529,7 @@ with tab_build:
         st.success(f"Model downloaded and saved to {config.model_path}! 🎉")
         st.rerun()
 
-# === Tab: Test Image Gen (existing) ===
+# === Tab: Test Image Gen ===
 with tab_imggen:
     st.header("Test Image Gen 🎨")
     all_files = get_gallery_files()
@@ -693,4 +693,4 @@ with tab_md_gallery:
             else:
                 st.warning("No MD files selected.")
     else:
-        st.warning("No MD files found.")
+        st.warning("No MD files found.")