Update app.py

app.py

@@ -26,10 +26,8 @@ from PIL import Image
 from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
 from typing import Optional
 
-# OpenAI client initialization
 client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'), organization=os.getenv('OPENAI_ORG_ID'))
 
-# Logging setup
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 log_records = []
@@ -38,7 +36,6 @@ class LogCaptureHandler(logging.Handler):
         log_records.append(record)
 logger.addHandler(LogCaptureHandler())
 
-# Streamlit configuration
 st.set_page_config(
     page_title="AI Vision & SFT Titans 🚀",
     page_icon="🤖",
@@ -51,7 +48,6 @@ st.set_page_config(
     }
 )
 
-# Session state initialization
 st.session_state.setdefault('history', [])
 st.session_state.setdefault('builder', None)
 st.session_state.setdefault('model_loaded', False)
@@ -64,6 +60,7 @@ st.session_state.setdefault('selected_model', "None")
 st.session_state.setdefault('cam0_file', None)
 st.session_state.setdefault('cam1_file', None)
 st.session_state.setdefault('characters', [])
+st.session_state.setdefault('char_form_reset', False)
 if 'asset_gallery_container' not in st.session_state:
     st.session_state['asset_gallery_container'] = st.sidebar.empty()
 
@@ -275,9 +272,7 @@ def process_text_with_prompt(text, prompt, model="gpt-4o-mini"):
     except Exception as e:
         return f"Error processing text with GPT: {str(e)}"
 
-# Character Editor Functions
 def randomize_character_content():
-    # Templates for randomizing Intro and Greeting
     intro_templates = [
         "{char} is a valiant knight who is silent and reserved, he looks handsome but aloof.",
         "{char} is a mischievous thief with a heart of gold, always sneaking around but helping those in need.",
@@ -297,7 +292,6 @@ def randomize_character_content():
     return intro, greeting
 
 def save_character(character_data):
-    # Save character to a JSON file
     characters = st.session_state.get('characters', [])
     characters.append(character_data)
     st.session_state['characters'] = characters
@@ -312,12 +306,10 @@ def load_characters():
     except FileNotFoundError:
         st.session_state['characters'] = []
 
-# Sidebar: Gallery Settings
 st.sidebar.subheader("Gallery Settings")
 st.session_state.setdefault('gallery_size', 2)
 st.session_state['gallery_size'] = st.sidebar.slider("Gallery Size", 1, 10, st.session_state['gallery_size'], key="gallery_size_slider")
 
-# Tabs setup
 tabs = st.tabs([
     "Camera Snap 📷", "Download PDFs 📥", "Test OCR 🔍", "Build Titan 🌱", 
     "Test Image Gen 🎨", "PDF Process 📄", "Image Process 🖼️", "MD Gallery 📚",
@@ -669,56 +661,65 @@ with tab_md_gallery:
 with tab_character_editor:
     st.header("Character Editor 🧑‍🎨")
     st.subheader("Create Your Character")
-    
-    # Character creation form with default values
-    if st.button("Randomize Content 🎲"):
-        name = f"Character_{random.randint(1000, 9999)}"
-        gender = random.choice(["Male", "Female", "Non-binary"])
-        intro, greeting = randomize_character_content()
-        st.session_state['char_name'] = name
-        st.session_state['char_gender'] = gender
-        st.session_state['char_intro'] = intro.format(char=name)
-        st.session_state['char_greeting'] = greeting.format(char=name)
-
-    name = st.text_input("Name (3-25 characters, letters, numbers, underscore, hyphen, space only)", 
-                         value=st.session_state.get('char_name', 'Character_3857'), 
-                         max_chars=25, 
-                         key="char_name")
-    gender = st.radio("Gender", ["Male", "Female", "Non-binary"], 
-                      index=["Male", "Female", "Non-binary"].index(st.session_state.get('char_gender', "Male")), 
-                      key="char_gender")
-    intro = st.text_area("Intro (Publicly seen)", 
-                         value=st.session_state.get('char_intro', 'Character_3857 is a gentle healer who speaks softly, always carrying herbs and a warm smile.'), 
-                         max_chars=300, 
-                         key="char_intro")
-    greeting = st.text_area("Greeting", 
-                            value=st.session_state.get('char_greeting', "A soft hand touches your shoulder. 'I’m Character_3857, here to heal your wounds—don’t worry, I’ve got you.'"), 
-                            max_chars=300, 
-                            key="char_greeting")
-
-    if st.button("Create Character", key="create_character"):
-        if not name or len(name) < 3:
-            st.error("Name must be 3-25 characters long.")
-        elif not re.match(r'^[a-zA-Z0-9 _-]+$', name):  # Fixed regex to avoid bad character range
-            st.error("Name can only contain letters, numbers, underscores, hyphens, and spaces.")
-        elif not intro or not greeting:
-            st.error("Intro and Greeting cannot be empty.")
+
+    with st.form(key="character_form"):
+        if st.session_state.get('char_form_reset', False):
+            default_name = ''
+            default_gender = "Male"
+            default_intro = ''
+            default_greeting = ''
+            st.session_state['char_form_reset'] = False
         else:
-            character_data = {
-                "name": name,
-                "gender": gender,
-                "intro": intro,
-                "greeting": greeting,
-                "created_at": datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
-                "tags": ["OC"]
-            }
-            save_character(character_data)
-            st.success(f"Character '{name}' created successfully!")
-            st.session_state['char_name'] = ''
-            st.session_state['char_gender'] = "Male"
-            st.session_state['char_intro'] = ''
-            st.session_state['char_greeting'] = ''
-            st.rerun()
+            default_name = 'Character_3857'
+            default_gender = "Male"
+            default_intro = 'Character_3857 is a gentle healer who speaks softly, always carrying herbs and a warm smile.'
+            default_greeting = "A soft hand touches your shoulder. 'I’m Character_3857, here to heal your wounds—don’t worry, I’ve got you.'"
+
+        if st.form_submit_button("Randomize Content 🎲"):
+            name = f"Character_{random.randint(1000, 9999)}"
+            gender = random.choice(["Male", "Female", "Non-binary"])
+            intro, greeting = randomize_character_content()
+            default_name = name
+            default_gender = gender
+            default_intro = intro.format(char=name)
+            default_greeting = greeting.format(char=name)
+
+        name = st.text_input("Name (3-25 characters, letters, numbers, underscore, hyphen, space only)", 
+                             value=default_name, 
+                             max_chars=25, 
+                             key="char_name")
+        gender = st.radio("Gender", ["Male", "Female", "Non-binary"], 
+                          index=["Male", "Female", "Non-binary"].index(default_gender), 
+                          key="char_gender")
+        intro = st.text_area("Intro (Publicly seen)", 
+                             value=default_intro, 
+                             max_chars=300, 
+                             key="char_intro")
+        greeting = st.text_area("Greeting", 
+                                value=default_greeting, 
+                                max_chars=300, 
+                                key="char_greeting")
+
+        if st.form_submit_button("Create Character"):
+            if not name or len(name) < 3:
+                st.error("Name must be 3-25 characters long.")
+            elif not re.match(r'^[a-zA-Z0-9 _-]+$', name):
+                st.error("Name can only contain letters, numbers, underscores, hyphens, and spaces.")
+            elif not intro or not greeting:
+                st.error("Intro and Greeting cannot be empty.")
+            else:
+                character_data = {
+                    "name": name,
+                    "gender": gender,
+                    "intro": intro,
+                    "greeting": greeting,
+                    "created_at": datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                    "tags": ["OC"]
+                }
+                save_character(character_data)
+                st.success(f"Character '{name}' created successfully!")
+                st.session_state['char_form_reset'] = True
+                st.rerun()
 
 with tab_character_gallery:
     st.header("Character Gallery 🖼️")