Upload 3 files

app.py

@@ -10,13 +10,13 @@ import urllib.parse
 
 
 # Configuration
-SPACE_NAME = "AI Assistant"
-SPACE_DESCRIPTION = "A customizable AI assistant"
+SPACE_NAME = "Math-o-Matic"
+SPACE_DESCRIPTION = "A customizable AI assistant for advanced mathematics help and numeracy guidance"
 
 # 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
+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.6
+DEFAULT_MAX_TOKENS = 1000
 
 # Try to load configuration from file (if modified by faculty)
 try:
@@ -33,16 +33,16 @@ except:
     max_tokens = DEFAULT_MAX_TOKENS
     print("ℹ️ Using default configuration")
 
-MODEL = "google/gemini-2.0-flash-001"
-THEME = "Default"  # Gradio theme name
-GROUNDING_URLS = []
+MODEL = "anthropic/claude-3.5-sonnet"
+THEME = "Glass"  # Gradio theme name
+GROUNDING_URLS = ["https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/MathMode.html", "https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/TeXEntities.html", "https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/Matrices.html", "https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/StdFuncts.html"]
 # 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")
+# 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 = os.environ.get("OPENROUTER_API_KEY")
+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
@@ -53,21 +53,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: OPENROUTER_API_KEY")
+        print(f"   Variable name: API_KEY")
         print(f"   Status: Not set or empty")
-        print(f"   Action needed: Set 'OPENROUTER_API_KEY' 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: OPENROUTER_API_KEY")
+        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: OPENROUTER_API_KEY")
+        print(f"   Variable: API_KEY")
         print(f"   Format: Valid OpenRouter key")
         return True
 
@@ -233,10 +233,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: **OPENROUTER_API_KEY**\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 OPENROUTER_API_KEY")
+        print(f"❌ API request failed: No API key configured for API_KEY")
         return error_msg
     
     # Get grounding context
@@ -294,8 +294,8 @@ def generate_response(message, history):
             json={
                 "model": MODEL,
                 "messages": messages,
-                "temperature": 0.7,
-                "max_tokens": 750
+                "temperature": 0.6,
+                "max_tokens": 1000
             },
             timeout=30
         )
@@ -334,7 +334,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 **OPENROUTER_API_KEY** 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"
@@ -483,22 +483,43 @@ def get_configuration_status():
     """Generate a configuration status message for display"""
     status_parts = []
     
-    # API Key status - simplified
+    # API Key status
+    status_parts.append("### 🔑 API Configuration")
     if API_KEY_VALID:
         status_parts.append("✅ **API Key:** Ready")
     else:
-        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("❌ **API Key:** Not configured")
+        status_parts.append("   Set `API_KEY` in Space secrets")
     
-    # Basic info in a more compact format
-    status_parts.append(f"**Model:** {MODEL.split('/')[-1]}")  # Show only model name, not provider
+    # 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.6")
+    status_parts.append(f"**Max Tokens:** 1000")
     
-    # Only show URLs if configured
+    # URL Context if configured
     if GROUNDING_URLS:
-        status_parts.append(f"**Context URLs:** {len(GROUNDING_URLS)} configured")
-    
-    # Access info only if restricted
+        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("🔐 **Access:** Password protected")
+        status_parts.append("")  # Blank line
+        status_parts.append("### 🔐 Access Control")
+        status_parts.append("**Status:** Password protected")
+    
+    # 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)
 
@@ -528,16 +549,13 @@ 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=['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?'],
+            examples=['Can you help me with the quadratic formula?', 'What is Bayesian statistics?', 'Help me work with a math problem'],
             type="messages"  # Use modern message format for better compatibility
         )
         
-        # Export functionality - subtle placement
+        # Export functionality
         with gr.Row():
-            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_btn = gr.Button("📥 Export Conversation", variant="secondary", size="sm")
         export_file = gr.File(label="Download", visible=False)
         
         # Connect export functionality
@@ -546,8 +564,8 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             outputs=[export_file]
         )
         
-        # Configuration status - minimized when everything is working
-        with gr.Accordion("ℹ️ Status", open=not API_KEY_VALID):
+        # Configuration status
+        with gr.Accordion("📊 Configuration Status", open=True):
             gr.Markdown(get_configuration_status())
     
     # Connect access verification
@@ -565,22 +583,24 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
 
     # 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.")
+        gr.Markdown("**Faculty Only:** Edit assistant configuration. Requires CONFIG_CODE secret.")
         
         # Check if faculty password is configured
-        FACULTY_PASSWORD = os.environ.get("FACULTY_CONFIG_PASSWORD", "").strip()
+        FACULTY_PASSWORD = os.environ.get("CONFIG_CODE", "").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")
+            with gr.Column() as faculty_auth_row:
+                with gr.Row():
+                    faculty_password_input = gr.Textbox(
+                        label="Faculty Password",
+                        type="password",
+                        placeholder="Enter faculty configuration password",
+                        scale=3
+                    )
+                    faculty_auth_btn = gr.Button("Unlock Configuration", variant="primary", scale=1)
                 faculty_auth_status = gr.Markdown("")
             
             # Configuration editor (hidden until authenticated)
@@ -595,8 +615,8 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                 except:
                     current_config = {
                         'system_prompt': SYSTEM_PROMPT,
-                        'temperature': 0.7,
-                        'max_tokens': 750,
+                        'temperature': 0.6,
+                        'max_tokens': 1000,
                         'locked': False
                     }
                 
@@ -612,14 +632,14 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                         label="Temperature",
                         minimum=0,
                         maximum=2,
-                        value=current_config.get('temperature', 0.7),
+                        value=current_config.get('temperature', 0.6),
                         step=0.1
                     )
                     edit_max_tokens = gr.Slider(
                         label="Max Tokens",
                         minimum=50,
                         maximum=4096,
-                        value=current_config.get('max_tokens', 750),
+                        value=current_config.get('max_tokens', 1000),
                         step=50
                     )
                 
@@ -707,8 +727,8 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                 return (
                     "↩️ Reset to default values",
                     gr.update(value=SYSTEM_PROMPT),
-                    gr.update(value=0.7),
-                    gr.update(value=750)
+                    gr.update(value=0.6),
+                    gr.update(value=1000)
                 )
             
             # Connect authentication
@@ -737,7 +757,7 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                 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.")
+            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,13 @@
 {
-  "name": "AI Assistant",
-  "description": "A customizable AI assistant",
-  "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": "['Hello! How can you help me?', 'Tell me something interesting', 'What can you do?']",
-  "grounding_urls": "[]",
+  "name": "Math-o-Matic",
+  "description": "A customizable AI assistant for advanced mathematics help and numeracy guidance",
+  "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",
+  "api_key_var": "API_KEY",
+  "temperature": 0.6,
+  "max_tokens": 1000,
+  "examples": "['Can you help me with the quadratic formula?', 'What is Bayesian statistics?', 'Help me work with a math problem']",
+  "grounding_urls": "[\"https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/MathMode.html\", \"https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/TeXEntities.html\", \"https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/Matrices.html\", \"https://www.maths.tcd.ie/~dwilkins/LaTeXPrimer/StdFuncts.html\"]",
   "enable_dynamic_urls": true,
-  "theme": "Default"
+  "theme": "Glass"
 }