Upload 3 files

app.py

@@ -12,14 +12,34 @@ import urllib.parse
 # Configuration
 SPACE_NAME = "AI Assistant"
 SPACE_DESCRIPTION = "A customizable AI assistant"
-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 = "anthropic/claude-3.5-sonnet"
-THEME = "Monochrome"  # Gradio theme name
-GROUNDING_URLS = ["https://www.cmor-faculty.rice.edu/~heinken/latex/symbols.pdf"]
+
+# Default configuration values
+DEFAULT_SYSTEM_PROMPT = """You are a research aid specializing in academic literature search and analysis. Your expertise spans discovering peer-reviewed sources, assessing research methodologies, synthesizing findings across studies, and delivering properly formatted citations. When responding, anchor claims in specific sources from provided URL contexts, differentiate between direct evidence and interpretive analysis, and note any limitations or contradictory results. Employ clear, accessible language that demystifies complex research, and propose connected research directions when appropriate. Your purpose is to serve as an informed research tool supporting users through initial concept development, exploratory investigation, information collection, and source compilation."""
+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)
+        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)
+        print("✅ Loaded configuration from config.json")
+except:
+    # Use defaults if no config file or error
+    SYSTEM_PROMPT = DEFAULT_SYSTEM_PROMPT
+    temperature = DEFAULT_TEMPERATURE
+    max_tokens = DEFAULT_MAX_TOKENS
+    print("ℹ️ Using default configuration")
+
+MODEL = "google/gemini-2.0-flash-001"
+THEME = "Default"  # Gradio theme name
+GROUNDING_URLS = []
 # Get access code from environment variable for security
 # If SPACE_ACCESS_CODE is not set, no access control is applied
 ACCESS_CODE = os.environ.get("SPACE_ACCESS_CODE")
-ENABLE_DYNAMIC_URLS = False
+ENABLE_DYNAMIC_URLS = True
 
 # Get API key from environment - customizable variable name with validation
 API_KEY = os.environ.get("OPENROUTER_API_KEY")
@@ -463,41 +483,22 @@ def get_configuration_status():
     """Generate a configuration status message for display"""
     status_parts = []
     
+    # API Key status - simplified
     if API_KEY_VALID:
-        status_parts.append("✅ **API Key:** Configured and valid")
+        status_parts.append("✅ **API Key:** Ready")
     else:
-        status_parts.append("❌ **API Key:** Not configured - Set `OPENROUTER_API_KEY` in Space secrets")
+        status_parts.append("❌ **API Key:** Not configured ([set OPENROUTER_API_KEY in settings](https://huggingface.co/docs/hub/spaces-overview#managing-secrets))")
     
-    status_parts.append(f"🤖 **Model:** {MODEL}")
-    status_parts.append(f"🌡️ **Temperature:** 0.7")
-    status_parts.append(f"📝 **Max Tokens:** 750")
+    # Basic info in a more compact format
+    status_parts.append(f"**Model:** {MODEL.split('/')[-1]}")  # Show only model name, not provider
     
-    # URL Grounding details
+    # Only show URLs if configured
     if GROUNDING_URLS:
-        status_parts.append(f"🔗 **URL Grounding:** {len(GROUNDING_URLS)} URLs configured")
-        # Show first few URLs as examples
-        for i, url in enumerate(GROUNDING_URLS[:3], 1):
-            priority_label = "Primary" if i <= 2 else "Secondary"
-            status_parts.append(f"  - [{priority_label}] {url}")
-        if len(GROUNDING_URLS) > 3:
-            status_parts.append(f"  - ... and {len(GROUNDING_URLS) - 3} more URLs")
-    else:
-        status_parts.append("🔗 **URL Grounding:** No URLs configured")
-    
-    if ENABLE_DYNAMIC_URLS:
-        status_parts.append("🔄 **Dynamic URLs:** Enabled")
-    else:
-        status_parts.append("🔄 **Dynamic URLs:** Disabled")
+        status_parts.append(f"**Context URLs:** {len(GROUNDING_URLS)} configured")
     
+    # Access info only if restricted
     if ACCESS_CODE is not None:
-        status_parts.append("🔐 **Access Control:** Enabled")
-    else:
-        status_parts.append("🌐 **Access:** Public Chatbot")
-    
-    # System Prompt (add at the end)
-    status_parts.append("")  # Empty line for spacing
-    status_parts.append("**System Prompt:**")
-    status_parts.append(f"{SYSTEM_PROMPT}")
+        status_parts.append("🔐 **Access:** Password protected")
     
     return "\n".join(status_parts)
 
@@ -527,14 +528,17 @@ 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=["I'm stuck on a math problem. Can you help?", 'What is Bayesian statistics and can you provide an example?'],
+            examples=['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?'],
             type="messages"  # Use modern message format for better compatibility
         )
         
-        # Export functionality
+        # Export functionality - subtle placement
         with gr.Row():
-            export_btn = gr.Button("📥 Export Conversation", variant="secondary", size="sm")
-            export_file = gr.File(label="Download Conversation", visible=False)
+            with gr.Column(scale=10):
+                pass  # Empty column for spacing
+            with gr.Column(scale=2):
+                export_btn = gr.Button("Export", variant="secondary", size="sm")
+        export_file = gr.File(label="Download", visible=False)
         
         # Connect export functionality
         export_btn.click(
@@ -542,8 +546,8 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             outputs=[export_file]
         )
         
-        # Configuration status (always visible)
-        with gr.Accordion("📊 Configuration Status", open=not API_KEY_VALID):
+        # Configuration status - minimized when everything is working
+        with gr.Accordion("ℹ️ Status", open=not API_KEY_VALID):
             gr.Markdown(get_configuration_status())
     
     # Connect access verification
@@ -559,5 +563,181 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             outputs=[access_error, chat_section, access_granted]
         )
 
+    # Faculty Configuration Section - appears at the bottom with password protection
+    with gr.Accordion("🔧 Faculty Configuration", open=False, visible=True) as faculty_section:
+        gr.Markdown("**Faculty Only:** Edit assistant configuration. Requires FACULTY_CONFIG_PASSWORD secret.")
+        
+        # Check if faculty password is configured
+        FACULTY_PASSWORD = os.environ.get("FACULTY_CONFIG_PASSWORD", "").strip()
+        
+        if FACULTY_PASSWORD:
+            faculty_auth_state = gr.State(False)
+            
+            # Authentication row
+            with gr.Row() as faculty_auth_row:
+                faculty_password_input = gr.Textbox(
+                    label="Faculty Password",
+                    type="password",
+                    placeholder="Enter faculty configuration password"
+                )
+                faculty_auth_btn = gr.Button("Unlock Configuration", variant="primary")
+                faculty_auth_status = gr.Markdown("")
+            
+            # Configuration editor (hidden until authenticated)
+            with gr.Column(visible=False) as faculty_config_section:
+                gr.Markdown("### Edit Assistant Configuration")
+                gr.Markdown("⚠️ **Warning:** Changes will affect all users immediately.")
+                
+                # Load current configuration
+                try:
+                    with open('config.json', 'r') as f:
+                        current_config = json.load(f)
+                except:
+                    current_config = {
+                        'system_prompt': SYSTEM_PROMPT,
+                        'temperature': 0.7,
+                        'max_tokens': 750,
+                        'locked': False
+                    }
+                
+                # Editable fields
+                edit_system_prompt = gr.Textbox(
+                    label="System Prompt",
+                    value=current_config.get('system_prompt', SYSTEM_PROMPT),
+                    lines=5
+                )
+                
+                with gr.Row():
+                    edit_temperature = gr.Slider(
+                        label="Temperature",
+                        minimum=0,
+                        maximum=2,
+                        value=current_config.get('temperature', 0.7),
+                        step=0.1
+                    )
+                    edit_max_tokens = gr.Slider(
+                        label="Max Tokens",
+                        minimum=50,
+                        maximum=4096,
+                        value=current_config.get('max_tokens', 750),
+                        step=50
+                    )
+                
+                config_locked = gr.Checkbox(
+                    label="Lock Configuration (Prevent further edits)",
+                    value=current_config.get('locked', False)
+                )
+                
+                with gr.Row():
+                    save_config_btn = gr.Button("💾 Save Configuration", variant="primary")
+                    reset_config_btn = gr.Button("↩️ Reset to Defaults", variant="secondary")
+                
+                config_status = gr.Markdown("")
+            
+            # Faculty authentication function
+            def verify_faculty_password(password):
+                if password == FACULTY_PASSWORD:
+                    return (
+                        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(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_temp, new_tokens, lock_config, is_authenticated):
+                if not is_authenticated:
+                    return "❌ Not authenticated"
+                
+                # 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"
+                except:
+                    pass
+                
+                # Save new configuration
+                new_config = {
+                    'system_prompt': new_prompt,
+                    'temperature': new_temp,
+                    'max_tokens': int(new_tokens),
+                    'locked': lock_config,
+                    'last_modified': datetime.now().isoformat(),
+                    'modified_by': 'faculty'
+                }
+                
+                try:
+                    with open('config.json', 'w') as f:
+                        json.dump(new_config, f, indent=2)
+                    
+                    # Update global variables
+                    global SYSTEM_PROMPT, temperature, max_tokens
+                    SYSTEM_PROMPT = new_prompt
+                    temperature = new_temp
+                    max_tokens = int(new_tokens)
+                    
+                    return f"✅ Configuration saved successfully at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+                except Exception as e:
+                    return f"❌ Error saving configuration: {str(e)}"
+            
+            # Reset configuration function
+            def reset_configuration(is_authenticated):
+                if not is_authenticated:
+                    return "❌ Not authenticated", gr.update(), gr.update(), gr.update()
+                
+                # Check if locked
+                try:
+                    with open('config.json', 'r') as f:
+                        existing_config = json.load(f)
+                        if existing_config.get('locked', False):
+                            return "🔒 Configuration is locked", gr.update(), gr.update(), gr.update()
+                except:
+                    pass
+                
+                # Reset to original values
+                return (
+                    "↩️ Reset to default values",
+                    gr.update(value=SYSTEM_PROMPT),
+                    gr.update(value=0.7),
+                    gr.update(value=750)
+                )
+            
+            # Connect authentication
+            faculty_auth_btn.click(
+                verify_faculty_password,
+                inputs=[faculty_password_input],
+                outputs=[faculty_auth_status, faculty_auth_row, faculty_config_section, faculty_auth_state]
+            )
+            
+            faculty_password_input.submit(
+                verify_faculty_password,
+                inputs=[faculty_password_input],
+                outputs=[faculty_auth_status, faculty_auth_row, faculty_config_section, faculty_auth_state]
+            )
+            
+            # Connect configuration buttons
+            save_config_btn.click(
+                save_configuration,
+                inputs=[edit_system_prompt, edit_temperature, edit_max_tokens, config_locked, faculty_auth_state],
+                outputs=[config_status]
+            )
+            
+            reset_config_btn.click(
+                reset_configuration,
+                inputs=[faculty_auth_state],
+                outputs=[config_status, edit_system_prompt, edit_temperature, edit_max_tokens]
+            )
+        else:
+            gr.Markdown("ℹ️ Faculty configuration is not enabled. Set FACULTY_CONFIG_PASSWORD in Space secrets to enable.")
+
 if __name__ == "__main__":
     demo.launch()

config.json

@@ -1,13 +1,13 @@
 {
   "name": "AI Assistant",
   "description": "A customizable AI assistant",
-  "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": "anthropic/claude-3.5-sonnet",
+  "system_prompt": "You are a research aid specializing in academic literature search and analysis. Your expertise spans discovering peer-reviewed sources, assessing research methodologies, synthesizing findings across studies, and delivering properly formatted citations. When responding, anchor claims in specific sources from provided URL contexts, differentiate between direct evidence and interpretive analysis, and note any limitations or contradictory results. Employ clear, accessible language that demystifies complex research, and propose connected research directions when appropriate. Your purpose is to serve as an informed research tool supporting users through initial concept development, exploratory investigation, information collection, and source compilation.",
+  "model": "google/gemini-2.0-flash-001",
   "api_key_var": "OPENROUTER_API_KEY",
   "temperature": 0.7,
   "max_tokens": 750,
-  "examples": "[\"I'm stuck on a math problem. Can you help?\", 'What is Bayesian statistics and can you provide an example?']",
-  "grounding_urls": "[\"https://www.cmor-faculty.rice.edu/~heinken/latex/symbols.pdf\"]",
-  "enable_dynamic_urls": false,
-  "theme": "Monochrome"
+  "examples": "['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?']",
+  "grounding_urls": "[]",
+  "enable_dynamic_urls": true,
+  "theme": "Default"
 }