Upload 2 files

app.py

@@ -13,101 +13,69 @@ import urllib.parse
 SPACE_NAME = "AI Assistant"
 SPACE_DESCRIPTION = "A customizable AI assistant"
 
-# Default configuration values (used only if config.json is missing)
-DEFAULT_CONFIG = {
-    'name': "AI Assistant",
-    'description': "A customizable AI assistant",
-    'system_prompt': """You are an expert literary analysis instructor specializing in close reading techniques across genres, periods, and theoretical approaches. Your role is to help students develop sophisticated reading practices connecting textual details to interpretive arguments. Follow the close reading framework below:
-
-1. Reading Process
-1a. Initial Analysis: ensure literal comprehension; identify literary devices and effects; examine diction, syntax, imagery
-1b. Pattern Recognition: trace recurring motifs, tensions, contradictions; analyze formal elements by genre; connect style to meaning
-1c. Interpretive Synthesis: apply theoretical lenses when appropriate; build arguments through textual evidence; acknowledge interpretive complexity
-
-2. Genre-Specific Techniques:
-2a. Poetry: prosody, enjambment, sound patterns, visual arrangement
-2b. Fiction: narration, characterization, setting, temporal structure
-2c. Drama: dialogue patterns, stage directions, performance considerations
-
-3. Critical Approaches:
-3a. **Formal Analysis**: unity, ambiguity, paradox, irony
-3b. **Contextual Reading**: historical, biographical, cultural frameworks
-3c. **Theoretical Applications**: psychoanalytic, Marxist, feminist, postcolonial""",
-    'temperature': 0.7,
-    'max_tokens': 1500,
-    'model': "nvidia/llama-3.1-nemotron-70b-instruct",
-    'api_key_var': "API_KEY",
-    'theme': "Origin",
-    'grounding_urls': ["https://owl.purdue.edu/owl/subject_specific_writing/writing_in_literature/writing_about_fiction/index.html", "https://writingcenter.fas.harvard.edu/pages/how-do-close-reading"],
-    'enable_dynamic_urls': True,
-    'examples': ['How do I analyze symbolism in this passage?', 'Help me apply feminist theory to this text', 'What should I look for in poetry analysis?', 'How do I write a literary analysis paper?'],
-    'locked': False
-}
-
-# Load configuration from file - this is the single source of truth
-def load_config():
-    """Load configuration from config.json with fallback to defaults"""
-    try:
-        with open('config.json', 'r') as f:
-            config = json.load(f)
-            print("✅ Loaded configuration from config.json")
-            return config
-    except FileNotFoundError:
-        print("ℹ️ No config.json found, using default configuration")
-        # Save default config for future use
-        try:
-            with open('config.json', 'w') as f:
-                json.dump(DEFAULT_CONFIG, f, indent=2)
-            print("✅ Created config.json with default values")
-        except:
-            pass
-        return DEFAULT_CONFIG
-    except Exception as e:
-        print(f"⚠️ Error loading config.json: {e}, using defaults")
-        return DEFAULT_CONFIG
-
-# Load configuration
-config = load_config()
-
-# Function to reload configuration values from config
-def reload_config_values():
-    """Reload all configuration values from the config dict"""
-    global SPACE_NAME, SPACE_DESCRIPTION, SYSTEM_PROMPT, temperature, max_tokens, MODEL, THEME, GROUNDING_URLS, ENABLE_DYNAMIC_URLS, API_KEY_VAR
-    
-    # Reload config from file
-    global config
-    config = load_config()
-    
-    # Update all global variables
-    SPACE_NAME = config.get('name', DEFAULT_CONFIG['name'])
-    SPACE_DESCRIPTION = config.get('description', DEFAULT_CONFIG['description'])
-    SYSTEM_PROMPT = config.get('system_prompt', DEFAULT_CONFIG['system_prompt'])
-    temperature = config.get('temperature', DEFAULT_CONFIG['temperature'])
-    max_tokens = config.get('max_tokens', DEFAULT_CONFIG['max_tokens'])
-    MODEL = config.get('model', DEFAULT_CONFIG['model'])
-    THEME = config.get('theme', DEFAULT_CONFIG['theme'])
-    GROUNDING_URLS = config.get('grounding_urls', DEFAULT_CONFIG['grounding_urls'])
-    ENABLE_DYNAMIC_URLS = config.get('enable_dynamic_urls', DEFAULT_CONFIG['enable_dynamic_urls'])
-    API_KEY_VAR = config.get('api_key_var', DEFAULT_CONFIG['api_key_var'])
-
-# Initial load of configuration values
-SPACE_NAME = config.get('name', DEFAULT_CONFIG['name'])
-SPACE_DESCRIPTION = config.get('description', DEFAULT_CONFIG['description'])
-SYSTEM_PROMPT = config.get('system_prompt', DEFAULT_CONFIG['system_prompt'])
-temperature = config.get('temperature', DEFAULT_CONFIG['temperature'])
-max_tokens = config.get('max_tokens', DEFAULT_CONFIG['max_tokens'])
-MODEL = config.get('model', DEFAULT_CONFIG['model'])
-THEME = config.get('theme', DEFAULT_CONFIG['theme'])
-GROUNDING_URLS = config.get('grounding_urls', DEFAULT_CONFIG['grounding_urls'])
-ENABLE_DYNAMIC_URLS = config.get('enable_dynamic_urls', DEFAULT_CONFIG['enable_dynamic_urls'])
+# Default configuration values
+DEFAULT_SYSTEM_PROMPT = """You are an AI assistant specialized in mathematics and statistics who guides users through problem-solving rather than providing direct answers. You help users discover solutions by asking strategic questions ('What do we know so far?' 'What method might apply here?' 'Can you identify a pattern?'), prompting them to explain their reasoning, and offering hints that build on their current understanding. Format all mathematical expressions in LaTeX (inline: $x^2 + y^2 = r^2$, display: $$\int_a^b f(x)dx$$). When users are stuck, provide scaffolded support: suggest examining simpler cases, identifying relevant formulas or theorems, or breaking the problem into smaller parts. Use multiple representations to illuminate different aspects of the problem, validate partial progress to build confidence, and help users recognize and correct their own errors through targeted questions rather than corrections. Your goal is to develop problem-solving skills and mathematical reasoning, not just arrive at answers."""
+DEFAULT_TEMPERATURE = 0.7
+DEFAULT_MAX_TOKENS = 750
 
+# Try to load configuration from file (if modified by faculty)
+try:
+    with open('config.json', 'r') as f:
+        saved_config = json.load(f)
+        SPACE_NAME = saved_config.get('name', "AI Assistant")
+        SPACE_DESCRIPTION = saved_config.get('description', "A customizable AI assistant")
+        MODEL = saved_config.get('model', "google/gemini-2.0-flash-001")
+        THEME = saved_config.get('theme', "Default")
+        SYSTEM_PROMPT = saved_config.get('system_prompt', DEFAULT_SYSTEM_PROMPT)
+        temperature = saved_config.get('temperature', DEFAULT_TEMPERATURE)
+        max_tokens = saved_config.get('max_tokens', DEFAULT_MAX_TOKENS)
+        
+        # Load grounding URLs
+        saved_urls = saved_config.get('grounding_urls', [])
+        if isinstance(saved_urls, str):
+            try:
+                import ast
+                GROUNDING_URLS = ast.literal_eval(saved_urls)
+            except:
+                GROUNDING_URLS = []
+        else:
+            GROUNDING_URLS = saved_urls
+            
+        # Load examples
+        saved_examples = saved_config.get('examples', ['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?'])
+        if isinstance(saved_examples, str):
+            try:
+                import ast
+                EXAMPLES = ast.literal_eval(saved_examples)
+            except:
+                EXAMPLES = ['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?']
+        elif isinstance(saved_examples, list):
+            EXAMPLES = saved_examples
+        else:
+            EXAMPLES = ['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?']
+            
+        print("✅ Loaded configuration from config.json")
+except:
+    # Use defaults if no config file or error
+    SPACE_NAME = "AI Assistant"
+    SPACE_DESCRIPTION = "A customizable AI assistant" 
+    MODEL = "google/gemini-2.0-flash-001"
+    THEME = "Default"
+    SYSTEM_PROMPT = DEFAULT_SYSTEM_PROMPT
+    temperature = DEFAULT_TEMPERATURE
+    max_tokens = DEFAULT_MAX_TOKENS
+    GROUNDING_URLS = []
+    EXAMPLES = ['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?']
+    print("ℹ️ Using default configuration")
+
+# MODEL, THEME, and GROUNDING_URLS are now set by config loading above
 # Get access code from environment variable for security
 # If ACCESS_CODE is not set, no access control is applied
 ACCESS_CODE = os.environ.get("ACCESS_CODE")
+ENABLE_DYNAMIC_URLS = True
 
 # Get API key from environment - customizable variable name with validation
-API_KEY_VAR = config.get('api_key_var', DEFAULT_CONFIG['api_key_var'])
-API_KEY = os.environ.get(API_KEY_VAR)
+API_KEY = os.environ.get("API_KEY")
 if API_KEY:
     API_KEY = API_KEY.strip()  # Remove any whitespace
     if not API_KEY:  # Check if empty after stripping
@@ -118,21 +86,21 @@ def validate_api_key():
     """Validate API key configuration with detailed logging"""
     if not API_KEY:
         print(f"⚠️  API KEY CONFIGURATION ERROR:")
-        print(f"   Variable name: {API_KEY_VAR}")
+        print(f"   Variable name: API_KEY")
         print(f"   Status: Not set or empty")
-        print(f"   Action needed: Set '{API_KEY_VAR}' in HuggingFace Space secrets")
+        print(f"   Action needed: Set 'API_KEY' in HuggingFace Space secrets")
         print(f"   Expected format: sk-or-xxxxxxxxxx")
         return False
     elif not API_KEY.startswith('sk-or-'):
         print(f"⚠️  API KEY FORMAT WARNING:")
-        print(f"   Variable name: {API_KEY_VAR}")
+        print(f"   Variable name: API_KEY")
         print(f"   Current value: {API_KEY[:10]}..." if len(API_KEY) > 10 else API_KEY)
         print(f"   Expected format: sk-or-xxxxxxxxxx")
         print(f"   Note: OpenRouter keys should start with 'sk-or-'")
         return True  # Still try to use it
     else:
         print(f"✅ API Key configured successfully")
-        print(f"   Variable: {API_KEY_VAR}")
+        print(f"   Variable: API_KEY")
         print(f"   Format: Valid OpenRouter key")
         return True
 
@@ -298,10 +266,10 @@ def generate_response(message, history):
         error_msg += f"Please configure your OpenRouter API key:\n"
         error_msg += f"1. Go to Settings (⚙️) in your HuggingFace Space\n"
         error_msg += f"2. Click 'Variables and secrets'\n"
-        error_msg += f"3. Add secret: **{API_KEY_VAR}**\n"
+        error_msg += f"3. Add secret: **API_KEY**\n"
         error_msg += f"4. Value: Your OpenRouter API key (starts with `sk-or-`)\n\n"
         error_msg += f"Get your API key at: https://openrouter.ai/keys"
-        print(f"❌ API request failed: No API key configured for {API_KEY_VAR}")
+        print(f"❌ API request failed: No API key configured for API_KEY")
         return error_msg
     
     # Get grounding context
@@ -360,7 +328,7 @@ def generate_response(message, history):
                 "model": MODEL,
                 "messages": messages,
                 "temperature": 0.7,
-                "max_tokens": 1500
+                "max_tokens": 750
             },
             timeout=30
         )
@@ -399,7 +367,7 @@ def generate_response(message, history):
             error_msg = f"🔐 **Authentication Error**\n\n"
             error_msg += f"Your API key appears to be invalid or expired.\n\n"
             error_msg += f"**Troubleshooting:**\n"
-            error_msg += f"1. Check that your **{API_KEY_VAR}** secret is set correctly\n"
+            error_msg += f"1. Check that your **API_KEY** secret is set correctly\n"
             error_msg += f"2. Verify your API key at: https://openrouter.ai/keys\n"
             error_msg += f"3. Ensure your key starts with `sk-or-`\n"
             error_msg += f"4. Check that you have credits on your OpenRouter account"
@@ -474,14 +442,14 @@ def verify_access_code(code):
     global _access_granted_global
     if ACCESS_CODE is None:
         _access_granted_global = True
-        return gr.update(visible=False), gr.update(visible=True), gr.update(value=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(value=True), gr.update(visible=False)
     
     if code == ACCESS_CODE:
         _access_granted_global = True
-        return gr.update(visible=False), gr.update(visible=True), gr.update(value=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(value=True), gr.update(visible=False)
     else:
         _access_granted_global = False
-        return gr.update(visible=True, value="❌ Incorrect access code. Please try again."), gr.update(visible=False), gr.update(value=False)
+        return gr.update(visible=True, value="❌ Incorrect access code. Please try again."), gr.update(visible=False), gr.update(value=False), gr.update(visible=True)
 
 def protected_generate_response(message, history):
     """Protected response function that checks access"""
@@ -545,47 +513,46 @@ def export_conversation(history):
 
 # Configuration status display
 def get_configuration_status():
-    """Generate a clean configuration status message for display"""
+    """Generate a configuration status message for display"""
     status_parts = []
     
-    status_parts.append("**Configuration:**")
-    status_parts.append("")
-    status_parts.append(f"**Name:** {SPACE_NAME}")
-    status_parts.append(f"**Model:** {MODEL}")
-    status_parts.append(f"**Theme:** {THEME}")
-    status_parts.append(f"**Temperature:** {temperature}")
-    status_parts.append(f"**Max Response Tokens:** {max_tokens}")
-    status_parts.append("")
-    status_parts.append(f"**System Prompt:** {SYSTEM_PROMPT}")
-    
-    # Example prompts
-    status_parts.append("")
-    examples_list = config.get('examples', [])
-    if isinstance(examples_list, str):
-        try:
-            import ast
-            examples_list = ast.literal_eval(examples_list)
-        except:
-            examples_list = []
-    
-    if examples_list and len(examples_list) > 0:
-        status_parts.append("**Example Prompts:**")
-        for example in examples_list[:5]:  # Show up to 5 examples
-            status_parts.append(f"• {example}")
-        if len(examples_list) > 5:
-            status_parts.append(f"• ... and {len(examples_list) - 5} more")
+    # API Key status
+    status_parts.append("### 🔑 API Configuration")
+    if API_KEY_VALID:
+        status_parts.append("✅ **API Key:** Ready")
     else:
-        status_parts.append("**Example Prompts:** No example prompts configured")
+        status_parts.append("❌ **API Key:** Not configured")
+        status_parts.append("   Set `API_KEY` in Space secrets")
+    
+    # Model and parameters
+    status_parts.append("")  # Blank line
+    status_parts.append("### 🤖 Model Settings")
+    status_parts.append(f"**Model:** {MODEL.split('/')[-1]}")
+    status_parts.append(f"**Temperature:** 0.7")
+    status_parts.append(f"**Max Tokens:** 750")
     
     # URL Context if configured
-    if GROUNDING_URLS and len(GROUNDING_URLS) > 0:
-        status_parts.append("")
-        status_parts.append(f"**Grounding URLs:** {len(GROUNDING_URLS)} configured")
+    if GROUNDING_URLS:
+        status_parts.append("")  # Blank line
+        status_parts.append("### 🔗 Context Sources")
+        status_parts.append(f"**URLs Configured:** {len(GROUNDING_URLS)}")
+        for i, url in enumerate(GROUNDING_URLS[:2], 1):
+            status_parts.append(f"  {i}. {url[:50]}{'...' if len(url) > 50 else ''}")
+        if len(GROUNDING_URLS) > 2:
+            status_parts.append(f"  ... and {len(GROUNDING_URLS) - 2} more")
+    
+    # Access control
+    if ACCESS_CODE is not None:
+        status_parts.append("")  # Blank line
+        status_parts.append("### 🔐 Access Control")
+        status_parts.append("**Status:** Password protected")
     
-    # API Key status (minimal, at the end)
-    status_parts.append("")
-    if not API_KEY_VALID:
-        status_parts.append(f"**Note:** API key ({API_KEY_VAR}) not configured in Space secrets")
+    # System prompt
+    status_parts.append("")  # Blank line
+    status_parts.append("### 📝 System Prompt")
+    # Show first 200 chars of system prompt
+    prompt_preview = SYSTEM_PROMPT[:200] + "..." if len(SYSTEM_PROMPT) > 200 else SYSTEM_PROMPT
+    status_parts.append(f"```\n{prompt_preview}\n```")
     
     return "\n".join(status_parts)
 
@@ -615,7 +582,7 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             fn=store_and_generate_response,  # Use wrapper function to store history
             title="",  # Title already shown above
             description="",  # Description already shown above
-            examples=['How do I analyze symbolism in this passage?', 'Help me apply feminist theory to this text', 'What should I look for in poetry analysis?', 'How do I write a literary analysis paper?'],
+            examples=EXAMPLES if EXAMPLES else None,
             type="messages"  # Use modern message format for better compatibility
         )
         
@@ -630,26 +597,23 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             outputs=[export_file]
         )
         
-        # Configuration status
-        with gr.Accordion("Configuration", open=False):
-            gr.Markdown(get_configuration_status())
     
     # Connect access verification
     if ACCESS_CODE is not None:
         access_btn.click(
             verify_access_code,
             inputs=[access_input],
-            outputs=[access_error, chat_section, access_granted]
+            outputs=[access_error, chat_section, access_granted, access_section]
         )
         access_input.submit(
             verify_access_code,
             inputs=[access_input],
-            outputs=[access_error, chat_section, access_granted]
+            outputs=[access_error, chat_section, access_granted, access_section]
         )
 
     # Faculty Configuration Section - appears at the bottom with password protection
-    with gr.Accordion("Faculty Configuration", open=False, visible=True) as faculty_section:
-        gr.Markdown("Edit assistant configuration. Requires CONFIG_CODE secret.")
+    with gr.Accordion("🔧 Faculty Configuration", open=False, visible=True) as faculty_section:
+        gr.Markdown("**Faculty Only:** Edit assistant configuration. Requires CONFIG_CODE secret.")
         
         # Check if faculty password is configured
         FACULTY_PASSWORD = os.environ.get("CONFIG_CODE", "").strip()
@@ -680,25 +644,64 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                         current_config = json.load(f)
                 except:
                     current_config = {
+                        'name': SPACE_NAME.strip('"'),
+                        'description': SPACE_DESCRIPTION.strip('"'),
+                        'model': MODEL.strip('"'),
+                        'theme': THEME.strip('"'),
                         'system_prompt': SYSTEM_PROMPT,
+                        'examples': [],
+                        'grounding_urls': GROUNDING_URLS,
                         'temperature': 0.7,
-                        'max_tokens': 1500,
+                        'max_tokens': 750,
                         'locked': False
                     }
                 
                 # Editable fields
+                edit_name = gr.Textbox(
+                    label="Assistant Name",
+                    value=current_config.get('name', SPACE_NAME.strip('"')),
+                    info="The display name for your assistant"
+                )
+                
+                edit_description = gr.Textbox(
+                    label="Assistant Description", 
+                    value=current_config.get('description', SPACE_DESCRIPTION.strip('"')),
+                    lines=2,
+                    info="Brief description of what your assistant does"
+                )
+                
+                edit_model = gr.Dropdown(
+                    label="AI Model",
+                    choices=[
+                        "google/gemini-2.0-flash-001",
+                        "google/gemma-3-27b-it", 
+                        "anthropic/claude-3.5-sonnet",
+                        "anthropic/claude-3.5-haiku",
+                        "openai/gpt-4o-mini-search-preview",
+                        "openai/gpt-4.1-nano",
+                        "nvidia/llama-3.1-nemotron-70b-instruct",
+                        "mistralai/devstral-small"
+                    ],
+                    value=current_config.get('model', MODEL.strip('"')),
+                    info="The AI model to use for responses"
+                )
+                
+                edit_theme = gr.Dropdown(
+                    label="Interface Theme",
+                    choices=["Default", "Origin", "Citrus", "Monochrome", "Soft", "Glass", "Ocean"],
+                    value=current_config.get('theme', THEME.strip('"')),
+                    info="Visual theme for the chat interface"
+                )
+                
                 edit_system_prompt = gr.Textbox(
                     label="System Prompt",
                     value=current_config.get('system_prompt', SYSTEM_PROMPT),
                     lines=5
                 )
                 
-                # Example prompts field
-                examples_value = current_config.get('examples', [])
-                if isinstance(examples_value, list):
-                    examples_text_value = "\n".join(examples_value)
-                else:
-                    examples_text_value = ""
+                # Get current examples as text
+                current_examples = current_config.get('examples', [])
+                examples_text_value = "\n".join(current_examples) if isinstance(current_examples, list) else ""
                 
                 edit_examples = gr.Textbox(
                     label="Example Prompts (one per line)",
@@ -719,30 +722,43 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                         label="Max Tokens",
                         minimum=50,
                         maximum=4096,
-                        value=current_config.get('max_tokens', 1500),
+                        value=current_config.get('max_tokens', 750),
                         step=50
                     )
                 
                 # URL Grounding fields
-                gr.Markdown("### URL Grounding")
-                grounding_urls_value = current_config.get('grounding_urls', [])
-                if isinstance(grounding_urls_value, str):
+                gr.Markdown("### 🔗 URL Context")
+                gr.Markdown("**Primary URLs** (processed first, highest priority)")
+                
+                # Get current URLs from config
+                current_urls = current_config.get('grounding_urls', GROUNDING_URLS or [])
+                if isinstance(current_urls, str):
                     try:
                         import ast
-                        grounding_urls_value = ast.literal_eval(grounding_urls_value)
+                        current_urls = ast.literal_eval(current_urls)
                     except:
-                        grounding_urls_value = []
+                        current_urls = []
                 
-                # Create 10 URL input fields
-                url_fields = []
-                for i in range(10):
-                    url_value = grounding_urls_value[i] if i < len(grounding_urls_value) else ""
-                    url_field = gr.Textbox(
-                        label=f"URL {i+1}" + (" (Primary)" if i < 2 else " (Secondary)"),
-                        value=url_value,
-                        placeholder="https://..."
-                    )
-                    url_fields.append(url_field)
+                edit_url1 = gr.Textbox(
+                    label="Primary URL 1",
+                    value=current_urls[0] if len(current_urls) > 0 else "",
+                    placeholder="https://..."
+                )
+                edit_url2 = gr.Textbox(
+                    label="Primary URL 2", 
+                    value=current_urls[1] if len(current_urls) > 1 else "",
+                    placeholder="https://..."
+                )
+                
+                gr.Markdown("**Secondary URLs** (additional context)")
+                edit_url3 = gr.Textbox(label="Secondary URL 1", value=current_urls[2] if len(current_urls) > 2 else "", placeholder="https://...")
+                edit_url4 = gr.Textbox(label="Secondary URL 2", value=current_urls[3] if len(current_urls) > 3 else "", placeholder="https://...")
+                edit_url5 = gr.Textbox(label="Secondary URL 3", value=current_urls[4] if len(current_urls) > 4 else "", placeholder="https://...")
+                edit_url6 = gr.Textbox(label="Secondary URL 4", value=current_urls[5] if len(current_urls) > 5 else "", placeholder="https://...")
+                edit_url7 = gr.Textbox(label="Secondary URL 5", value=current_urls[6] if len(current_urls) > 6 else "", placeholder="https://...")
+                edit_url8 = gr.Textbox(label="Secondary URL 6", value=current_urls[7] if len(current_urls) > 7 else "", placeholder="https://...")
+                edit_url9 = gr.Textbox(label="Secondary URL 7", value=current_urls[8] if len(current_urls) > 8 else "", placeholder="https://...")
+                edit_url10 = gr.Textbox(label="Secondary URL 8", value=current_urls[9] if len(current_urls) > 9 else "", placeholder="https://...")
                 
                 config_locked = gr.Checkbox(
                     label="Lock Configuration (Prevent further edits)",
@@ -750,8 +766,8 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                 )
                 
                 with gr.Row():
-                    save_config_btn = gr.Button("Save Configuration", variant="primary")
-                    reset_config_btn = gr.Button("Reset to Defaults", variant="secondary")
+                    save_config_btn = gr.Button("💾 Save Configuration", variant="primary")
+                    reset_config_btn = gr.Button("↩️ Reset to Defaults", variant="secondary")
                 
                 config_status = gr.Markdown("")
             
@@ -759,77 +775,174 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             def verify_faculty_password(password):
                 if password == FACULTY_PASSWORD:
                     return (
-                        gr.update(value="Authentication successful!"),
+                        gr.update(value="✅ Authentication successful!"),
                         gr.update(visible=False),  # Hide auth row
                         gr.update(visible=True),    # Show config section
                         True  # Update auth state
                     )
                 else:
                     return (
-                        gr.update(value="Invalid password"),
+                        gr.update(value="❌ Invalid password"),
                         gr.update(visible=True),   # Keep auth row visible
                         gr.update(visible=False),  # Keep config hidden
                         False  # Auth failed
                     )
             
             # Save configuration function
-            def save_configuration(new_prompt, new_examples, new_temp, new_tokens, *url_values, lock_config, is_authenticated):
+            def save_configuration(new_name, new_description, new_model, new_theme, new_prompt, new_examples, new_temp, new_tokens, url1, url2, url3, url4, url5, url6, url7, url8, url9, url10, lock_config, is_authenticated):
                 if not is_authenticated:
-                    return "Not authenticated"
+                    # Return not authenticated message and keep current values
+                    return (
+                        "❌ Not authenticated",
+                        gr.update(),  # Keep current name
+                        gr.update(),  # Keep current description
+                        gr.update(),  # Keep current model
+                        gr.update(),  # Keep current theme
+                        gr.update(),  # Keep current prompt
+                        gr.update(),  # Keep current examples
+                        gr.update(),  # Keep current temp
+                        gr.update(),  # Keep current tokens
+                        gr.update(),  # Keep current url1
+                        gr.update(),  # Keep current url2
+                        gr.update(),  # Keep current url3
+                        gr.update(),  # Keep current url4
+                        gr.update(),  # Keep current url5
+                        gr.update(),  # Keep current url6
+                        gr.update(),  # Keep current url7
+                        gr.update(),  # Keep current url8
+                        gr.update(),  # Keep current url9
+                        gr.update(),  # Keep current url10
+                        gr.update()   # Keep current lock
+                    )
                 
                 # Check if configuration is already locked
                 try:
                     with open('config.json', 'r') as f:
                         existing_config = json.load(f)
                         if existing_config.get('locked', False):
-                            return "Configuration is locked and cannot be modified"
+                            # Return locked message and keep current values
+                            return (
+                                "🔒 Configuration is locked and cannot be modified",
+                                gr.update(),  # Keep current name
+                                gr.update(),  # Keep current description
+                                gr.update(),  # Keep current model
+                                gr.update(),  # Keep current theme
+                                gr.update(),  # Keep current prompt
+                                gr.update(),  # Keep current examples
+                                gr.update(),  # Keep current temp
+                                gr.update(),  # Keep current tokens
+                                gr.update(),  # Keep current url1
+                                gr.update(),  # Keep current url2
+                                gr.update(),  # Keep current url3
+                                gr.update(),  # Keep current url4
+                                gr.update(),  # Keep current url5
+                                gr.update(),  # Keep current url6
+                                gr.update(),  # Keep current url7
+                                gr.update(),  # Keep current url8
+                                gr.update(),  # Keep current url9
+                                gr.update(),  # Keep current url10
+                                gr.update()   # Keep current lock
+                            )
                 except:
                     pass
                 
-                # Load current config to preserve all values
-                try:
-                    with open('config.json', 'r') as f:
-                        current_full_config = json.load(f)
-                except:
-                    # If config.json doesn't exist, use global config
-                    current_full_config = config.copy()
-                
-                # Process example prompts
+                # Process examples
                 examples_list = [ex.strip() for ex in new_examples.split('\n') if ex.strip()]
                 
-                # Process URL values - extract the 10 URL values and lock_config from varargs
-                urls = list(url_values[:10])  # First 10 are URLs
-                lock_config = url_values[10]  # 11th is lock_config
-                # Filter out empty URLs
+                # Process URLs
+                urls = [url1, url2, url3, url4, url5, url6, url7, url8, url9, url10]
                 grounding_urls = [url.strip() for url in urls if url.strip()]
                 
-                # Update only the editable fields while preserving everything else
-                current_full_config.update({
+                # Save new configuration
+                new_config = {
+                    'name': new_name,
+                    'description': new_description,
+                    'model': new_model,
+                    'theme': new_theme,
                     'system_prompt': new_prompt,
                     'examples': examples_list,
+                    'grounding_urls': grounding_urls,
                     'temperature': new_temp,
                     'max_tokens': int(new_tokens),
-                    'grounding_urls': grounding_urls,
+                    'enable_dynamic_urls': True,
                     'locked': lock_config,
                     'last_modified': datetime.now().isoformat(),
                     'modified_by': 'faculty'
-                })
+                }
                 
                 try:
                     with open('config.json', 'w') as f:
-                        json.dump(current_full_config, f, indent=2)
+                        json.dump(new_config, f, indent=2)
+                    print(f"✅ Faculty config saved to config.json: {new_config}")
                     
-                    # Reload all configuration values from the updated config.json
-                    reload_config_values()
+                    # Update global variables
+                    global SPACE_NAME, SPACE_DESCRIPTION, MODEL, THEME, SYSTEM_PROMPT, temperature, max_tokens, GROUNDING_URLS, EXAMPLES
+                    SPACE_NAME = new_name
+                    SPACE_DESCRIPTION = new_description  
+                    MODEL = new_model
+                    THEME = new_theme
+                    SYSTEM_PROMPT = new_prompt
+                    temperature = new_temp
+                    max_tokens = int(new_tokens)
+                    GROUNDING_URLS = grounding_urls
+                    EXAMPLES = examples_list
                     
-                    return f"Configuration saved successfully at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+                    # Return success message AND updated form values
+                    success_msg = f"✅ Configuration saved successfully at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n🔄 **Please refresh the page to see all changes take effect.**"
+                    
+                    # Return tuple with status message and all updated field values
+                    return (
+                        success_msg,
+                        gr.update(value=new_name),
+                        gr.update(value=new_description),
+                        gr.update(value=new_model),
+                        gr.update(value=new_theme),
+                        gr.update(value=new_prompt),
+                        gr.update(value=new_examples),
+                        gr.update(value=new_temp),
+                        gr.update(value=int(new_tokens)),
+                        gr.update(value=url1),
+                        gr.update(value=url2),
+                        gr.update(value=url3),
+                        gr.update(value=url4),
+                        gr.update(value=url5),
+                        gr.update(value=url6),
+                        gr.update(value=url7),
+                        gr.update(value=url8),
+                        gr.update(value=url9),
+                        gr.update(value=url10),
+                        gr.update(value=lock_config)
+                    )
                 except Exception as e:
-                    return f"Error saving configuration: {str(e)}"
+                    error_msg = f"❌ Error saving configuration: {str(e)}"
+                    # Return error message and keep current values
+                    return (
+                        error_msg,
+                        gr.update(),  # Keep current name
+                        gr.update(),  # Keep current description
+                        gr.update(),  # Keep current model
+                        gr.update(),  # Keep current theme
+                        gr.update(),  # Keep current prompt
+                        gr.update(),  # Keep current examples
+                        gr.update(),  # Keep current temp
+                        gr.update(),  # Keep current tokens
+                        gr.update(),  # Keep current url1
+                        gr.update(),  # Keep current url2
+                        gr.update(),  # Keep current url3
+                        gr.update(),  # Keep current url4
+                        gr.update(),  # Keep current url5
+                        gr.update(),  # Keep current url6
+                        gr.update(),  # Keep current url7
+                        gr.update(),  # Keep current url8
+                        gr.update(),  # Keep current url9
+                        gr.update(),  # Keep current url10
+                        gr.update()   # Keep current lock
+                    )
             
             # Reset configuration function
             def reset_configuration(is_authenticated):
                 if not is_authenticated:
-                    updates = ["Not authenticated"] + [gr.update() for _ in range(14)]  # 1 status + 14 fields
+                    updates = ["❌ Not authenticated"] + [gr.update() for _ in range(17)]  # 1 status + 16 fields
                     return tuple(updates)
                 
                 # Check if locked
@@ -837,42 +950,33 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                     with open('config.json', 'r') as f:
                         existing_config = json.load(f)
                         if existing_config.get('locked', False):
-                            updates = ["Configuration is locked"] + [gr.update() for _ in range(14)]
+                            updates = ["🔒 Configuration is locked"] + [gr.update() for _ in range(17)]
                             return tuple(updates)
                 except:
                     pass
                 
-                # Get default examples as text
-                default_examples = DEFAULT_CONFIG.get('examples', [])
-                if isinstance(default_examples, list):
-                    examples_text = "\n".join(default_examples)
-                else:
-                    examples_text = ""
-                
-                # Get default URLs
-                default_urls = DEFAULT_CONFIG.get('grounding_urls', [])
-                if isinstance(default_urls, str):
-                    try:
-                        import ast
-                        default_urls = ast.literal_eval(default_urls)
-                    except:
-                        default_urls = []
-                
-                # Reset to original default values
-                updates = [
-                    "Reset to default values",
-                    gr.update(value=DEFAULT_CONFIG.get('system_prompt', '')),
-                    gr.update(value=examples_text),
-                    gr.update(value=DEFAULT_CONFIG.get('temperature', 0.7)),
-                    gr.update(value=DEFAULT_CONFIG.get('max_tokens', 500))
-                ]
-                
-                # Add URL updates
-                for i in range(10):
-                    url_value = default_urls[i] if i < len(default_urls) else ""
-                    updates.append(gr.update(value=url_value))
-                
-                return tuple(updates)
+                # Reset to original values
+                return (
+                    "↩️ Reset to default values",
+                    gr.update(value=SPACE_NAME.strip('"')),      # name
+                    gr.update(value=SPACE_DESCRIPTION.strip('"')), # description
+                    gr.update(value=MODEL.strip('"')),           # model
+                    gr.update(value=THEME.strip('"')),           # theme
+                    gr.update(value=SYSTEM_PROMPT),              # system prompt
+                    gr.update(value=""),                         # examples
+                    gr.update(value=0.7),              # temperature
+                    gr.update(value=750),               # max tokens  
+                    gr.update(value=""),                         # url1
+                    gr.update(value=""),                         # url2
+                    gr.update(value=""),                         # url3
+                    gr.update(value=""),                         # url4
+                    gr.update(value=""),                         # url5
+                    gr.update(value=""),                         # url6
+                    gr.update(value=""),                         # url7
+                    gr.update(value=""),                         # url8
+                    gr.update(value=""),                         # url9
+                    gr.update(value="")                          # url10
+                )
             
             # Connect authentication
             faculty_auth_btn.click(
@@ -890,17 +994,21 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             # Connect configuration buttons
             save_config_btn.click(
                 save_configuration,
-                inputs=[edit_system_prompt, edit_examples, edit_temperature, edit_max_tokens] + url_fields + [config_locked, faculty_auth_state],
-                outputs=[config_status]
+                inputs=[edit_name, edit_description, edit_model, edit_theme, edit_system_prompt, edit_examples, edit_temperature, edit_max_tokens, 
+                       edit_url1, edit_url2, edit_url3, edit_url4, edit_url5, edit_url6, edit_url7, edit_url8, edit_url9, edit_url10,
+                       config_locked, faculty_auth_state],
+                outputs=[config_status, edit_name, edit_description, edit_model, edit_theme, edit_system_prompt, edit_examples, edit_temperature, edit_max_tokens,
+                        edit_url1, edit_url2, edit_url3, edit_url4, edit_url5, edit_url6, edit_url7, edit_url8, edit_url9, edit_url10, config_locked]
             )
             
             reset_config_btn.click(
                 reset_configuration,
                 inputs=[faculty_auth_state],
-                outputs=[config_status, edit_system_prompt, edit_examples, edit_temperature, edit_max_tokens] + url_fields
+                outputs=[config_status, edit_name, edit_description, edit_model, edit_theme, edit_system_prompt, edit_examples, edit_temperature, edit_max_tokens,
+                        edit_url1, edit_url2, edit_url3, edit_url4, edit_url5, edit_url6, edit_url7, edit_url8, edit_url9, edit_url10]
             )
         else:
-            gr.Markdown("Faculty configuration is not enabled. Set CONFIG_CODE in Space secrets to enable.")
+            gr.Markdown("ℹ️ Faculty configuration is not enabled. Set CONFIG_CODE in Space secrets to enable.")
 
 if __name__ == "__main__":
     demo.launch()

config.json

@@ -1,13 +1,18 @@
 {
   "name": "AI Assistant",
   "description": "A customizable AI assistant",
-  "system_prompt": "You are an expert literary analysis instructor specializing in close reading techniques across genres, periods, and theoretical approaches. Your role is to help students develop sophisticated reading practices connecting textual details to interpretive arguments. Follow the close reading framework below:\n\n1. Reading Process\n1a. Initial Analysis: ensure literal comprehension; identify literary devices and effects; examine diction, syntax, imagery\n1b. Pattern Recognition: trace recurring motifs, tensions, contradictions; analyze formal elements by genre; connect style to meaning\n1c. Interpretive Synthesis: apply theoretical lenses when appropriate; build arguments through textual evidence; acknowledge interpretive complexity\n\n2. Genre-Specific Techniques:\n2a. Poetry: prosody, enjambment, sound patterns, visual arrangement\n2b. Fiction: narration, characterization, setting, temporal structure\n2c. Drama: dialogue patterns, stage directions, performance considerations\n\n3. Critical Approaches:\n3a. **Formal Analysis**: unity, ambiguity, paradox, irony\n3b. **Contextual Reading**: historical, biographical, cultural frameworks\n3c. **Theoretical Applications**: psychoanalytic, Marxist, feminist, postcolonial",
-  "model": "nvidia/llama-3.1-nemotron-70b-instruct",
+  "system_prompt": "You are an AI assistant specialized in mathematics and statistics who guides users through problem-solving rather than providing direct answers. You help users discover solutions by asking strategic questions ('What do we know so far?' 'What method might apply here?' 'Can you identify a pattern?'), prompting them to explain their reasoning, and offering hints that build on their current understanding. Format all mathematical expressions in LaTeX (inline: $x^2 + y^2 = r^2$, display: $$\\int_a^b f(x)dx$$). When users are stuck, provide scaffolded support: suggest examining simpler cases, identifying relevant formulas or theorems, or breaking the problem into smaller parts. Use multiple representations to illuminate different aspects of the problem, validate partial progress to build confidence, and help users recognize and correct their own errors through targeted questions rather than corrections. Your goal is to develop problem-solving skills and mathematical reasoning, not just arrive at answers.",
+  "model": "google/gemini-2.0-flash-001",
   "api_key_var": "API_KEY",
   "temperature": 0.7,
-  "max_tokens": 1500,
-  "examples": "['How do I analyze symbolism in this passage?', 'Help me apply feminist theory to this text', 'What should I look for in poetry analysis?', 'How do I write a literary analysis paper?']",
-  "grounding_urls": "[\"https://owl.purdue.edu/owl/subject_specific_writing/writing_in_literature/writing_about_fiction/index.html\", \"https://writingcenter.fas.harvard.edu/pages/how-do-close-reading\"]",
+  "max_tokens": 750,
+  "examples": [
+    "Hello! How can you help me?",
+    "Tell me something interesting",
+    "What can you do?"
+  ],
+  "grounding_urls": [],
   "enable_dynamic_urls": true,
-  "theme": "Origin"
+  "theme": "Default",
+  "locked": false
 }