Upload 3 files

app.py

@@ -10,8 +10,8 @@ import urllib.parse
 
 
 # Configuration
-SPACE_NAME = "CUNY Virgil"
-SPACE_DESCRIPTION = "Adaptable AI assistant for research methods support"
+SPACE_NAME = "Virgil Research Guide"
+SPACE_DESCRIPTION = "A customizable AI assistant"
 
 # 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."""
@@ -33,9 +33,9 @@ except:
     max_tokens = DEFAULT_MAX_TOKENS
     print("ℹ️ Using default configuration")
 
-MODEL = "nvidia/llama-3.1-nemotron-70b-instruct"
-THEME = "Glass"  # Gradio theme name
-GROUNDING_URLS = []
+MODEL = "google/gemini-2.0-flash-001"
+THEME = "Default"  # Gradio theme name
+GROUNDING_URLS = ["https://bcc-cuny.libguides.com/", "https://researchguides.journalism.cuny.edu/"]
 # 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")
@@ -549,7 +549,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=['Help me narrow a research question'],
+            examples=['Help me prepare for ethnographic fieldwork?', 'What are some methods of digital ethnography?'],
             type="messages"  # Use modern message format for better compatibility
         )
         
@@ -611,37 +611,40 @@ 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': 750,
                         'locked': False
                     }
                 
-                # Editable fields
+                # Editable fields - Order matches the Configuration tab
+                # 1. Assistant Identity
                 edit_name = gr.Textbox(
                     label="Assistant Name",
-                    value=current_config.get('name', SPACE_NAME.strip('"')),
-                    info="The display name for your assistant"
+                    value=current_config.get('name', SPACE_NAME),
+                    placeholder="My AI Assistant"
                 )
                 
                 edit_description = gr.Textbox(
-                    label="Assistant Description", 
-                    value=current_config.get('description', SPACE_DESCRIPTION.strip('"')),
+                    label="Assistant Description",
+                    value=current_config.get('description', SPACE_DESCRIPTION),
                     lines=2,
-                    info="Brief description of what your assistant does"
+                    placeholder="A helpful AI assistant for..."
                 )
                 
+                # 2. System Prompt
+                edit_system_prompt = gr.Textbox(
+                    label="System Prompt",
+                    value=current_config.get('system_prompt', SYSTEM_PROMPT),
+                    lines=5
+                )
+                
+                # 3. Model Selection
                 edit_model = gr.Dropdown(
-                    label="AI Model",
+                    label="Model",
                     choices=[
                         "google/gemini-2.0-flash-001",
-                        "google/gemma-3-27b-it", 
+                        "google/gemma-3-27b-it",
                         "anthropic/claude-3.5-sonnet",
                         "anthropic/claude-3.5-haiku",
                         "openai/gpt-4o-mini-search-preview",
@@ -649,26 +652,15 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                         "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"
+                    value=current_config.get('model', MODEL)
                 )
                 
-                edit_system_prompt = gr.Textbox(
-                    label="System Prompt",
-                    value=current_config.get('system_prompt', SYSTEM_PROMPT),
-                    lines=5
-                )
-                
-                # Get current examples as text
-                current_examples = current_config.get('examples', [])
-                examples_text_value = "\n".join(current_examples) if isinstance(current_examples, list) else ""
+                # 4. Example Prompts
+                examples_value = current_config.get('examples', [])
+                if isinstance(examples_value, list):
+                    examples_text_value = "\n".join(examples_value)
+                else:
+                    examples_text_value = ""
                 
                 edit_examples = gr.Textbox(
                     label="Example Prompts (one per line)",
@@ -677,6 +669,7 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                     placeholder="What can you help me with?\nExplain this concept\nHelp me understand..."
                 )
                 
+                # 5. Model Parameters
                 with gr.Row():
                     edit_temperature = gr.Slider(
                         label="Temperature",
@@ -693,39 +686,26 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                         step=50
                     )
                 
-                # URL Grounding fields
-                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):
+                # 6. URL Grounding
+                gr.Markdown("### URL Grounding")
+                grounding_urls_value = current_config.get('grounding_urls', [])
+                if isinstance(grounding_urls_value, str):
                     try:
                         import ast
-                        current_urls = ast.literal_eval(current_urls)
+                        grounding_urls_value = ast.literal_eval(grounding_urls_value)
                     except:
-                        current_urls = []
-                
-                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://..."
-                )
+                        grounding_urls_value = []
                 
-                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://...")
+                # 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)
                 
                 config_locked = gr.Checkbox(
                     label="Lock Configuration (Prevent further edits)",
@@ -756,7 +736,7 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                     )
             
             # Save configuration function
-            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):
+            def save_configuration(new_name, new_description, new_prompt, new_model, new_examples, new_temp, new_tokens, *url_values, lock_config, is_authenticated):
                 if not is_authenticated:
                     return "❌ Not authenticated"
                 
@@ -769,53 +749,53 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
                 except:
                     pass
                 
-                # Process examples
+                # 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
                 examples_list = [ex.strip() for ex in new_examples.split('\n') if ex.strip()]
                 
-                # Process URLs
-                urls = [url1, url2, url3, url4, url5, url6, url7, url8, url9, url10]
+                # Process URL values - lock_config is the last parameter
+                urls = list(url_values[:-1])  # All but last are URLs
+                lock_config_from_args = url_values[-1]  # Last is lock_config
+                # Filter out empty URLs
                 grounding_urls = [url.strip() for url in urls if url.strip()]
                 
-                # Save new configuration
-                new_config = {
+                # Update all editable fields while preserving everything else
+                current_full_config.update({
                     'name': new_name,
                     'description': new_description,
-                    'model': new_model,
-                    'theme': new_theme,
                     'system_prompt': new_prompt,
+                    'model': new_model,
                     'examples': examples_list,
-                    'grounding_urls': grounding_urls,
                     'temperature': new_temp,
                     'max_tokens': int(new_tokens),
-                    'enable_dynamic_urls': True,
-                    'locked': lock_config,
+                    'grounding_urls': grounding_urls,
+                    'locked': lock_config_from_args,
                     'last_modified': datetime.now().isoformat(),
-                    'modified_by': 'faculty'
+                    'last_modified_by': 'faculty'
                 }
                 
                 try:
                     with open('config.json', 'w') as f:
-                        json.dump(new_config, f, indent=2)
+                        json.dump(current_full_config, f, indent=2)
                     
-                    # Update global variables
-                    global SPACE_NAME, SPACE_DESCRIPTION, MODEL, THEME, SYSTEM_PROMPT, temperature, max_tokens, GROUNDING_URLS
-                    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
+                    # Reload all configuration values
+                    reload_config_values()
                     
-                    return f"✅ Configuration saved successfully at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n⚠️ **Note:** Example prompts will update on next page refresh."
+                    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:
-                    updates = ["❌ Not authenticated"] + [gr.update() for _ in range(17)]  # 1 status + 16 fields
+                    updates = ["❌ Not authenticated"] + [gr.update() for _ in range(16)]  # 1 status + 16 fields
                     return tuple(updates)
                 
                 # Check if locked
@@ -823,33 +803,45 @@ 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(17)]
+                            updates = ["🔒 Configuration is locked"] + [gr.update() for _ in range(16)]
                             return tuple(updates)
                 except:
                     pass
                 
-                # 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
-                )
+                # 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('name', SPACE_NAME)),
+                    gr.update(value=DEFAULT_CONFIG.get('description', SPACE_DESCRIPTION)),
+                    gr.update(value=DEFAULT_CONFIG.get('system_prompt', SYSTEM_PROMPT)),
+                    gr.update(value=DEFAULT_CONFIG.get('model', MODEL)),
+                    gr.update(value=examples_text),
+                    gr.update(value=DEFAULT_CONFIG.get('temperature', 0.7)),
+                    gr.update(value=DEFAULT_CONFIG.get('max_tokens', 750))
+                ]
+                
+                # 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)
             
             # Connect authentication
             faculty_auth_btn.click(
@@ -867,17 +859,14 @@ with gr.Blocks(title=SPACE_NAME, theme=theme_class()) as demo:
             # Connect configuration buttons
             save_config_btn.click(
                 save_configuration,
-                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],
+                inputs=[edit_name, edit_description, edit_system_prompt, edit_model, edit_examples, edit_temperature, edit_max_tokens] + url_fields + [config_locked, faculty_auth_state],
                 outputs=[config_status]
             )
             
             reset_config_btn.click(
                 reset_configuration,
                 inputs=[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]
+                outputs=[config_status, edit_name, edit_description, edit_system_prompt, edit_model, edit_examples, edit_temperature, edit_max_tokens] + url_fields
             )
         else:
             gr.Markdown("ℹ️ Faculty configuration is not enabled. Set CONFIG_CODE in Space secrets to enable.")

config.json

@@ -1,13 +1,13 @@
 {
-  "name": "CUNY Virgil",
-  "description": "Adaptable AI assistant for research methods support",
+  "name": "Virgil Research Guide",
+  "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": "nvidia/llama-3.1-nemotron-70b-instruct",
+  "model": "google/gemini-2.0-flash-001",
   "api_key_var": "API_KEY",
   "temperature": 0.7,
   "max_tokens": 750,
-  "examples": "['Help me narrow a research question']",
-  "grounding_urls": "[]",
+  "examples": "['Help me prepare for ethnographic fieldwork?', 'What are some methods of digital ethnography?']",
+  "grounding_urls": "[\"https://bcc-cuny.libguides.com/\", \"https://researchguides.journalism.cuny.edu/\"]",
   "enable_dynamic_urls": true,
-  "theme": "Glass"
+  "theme": "Default"
 }