Update app.py

app.py

@@ -11,31 +11,9 @@ def list_saved_results():
     result_files = [f for f in os.listdir(results_dir) if f.endswith(".pkl")]
     # Sort by modification time (newest first)
     result_files.sort(key=lambda x: os.path.getmtime(os.path.join(results_dir, x)), reverse=True)
-    
-    # Format the filenames for display
     formatted_results = []
     for filename in result_files:
-        # Extract date from filename
-        parts = filename.split("_")
-        if len(parts) >= 2:
-            # Try to make the filename more readable
-            try:
-                # Assuming format is model1_vs_model2_dataset_YYYYMMDD_HHMMSS.pkl
-                models_part = "_".join(parts[:-2])  # Everything except timestamp
-                date_part = parts[-2]
-                time_part = parts[-1].replace(".pkl", "")
-                
-                # Format date for display
-                date_str = f"{date_part[:4]}-{date_part[4:6]}-{date_part[6:]}"
-                time_str = f"{time_part[:2]}:{time_part[2:4]}:{time_part[4:]}"
-                
-                display_name = f"{models_part} ({date_str} {time_str})"
-                formatted_results.append((display_name, filename))
-            except:
-                # If parsing fails, just use the filename
-                formatted_results.append((filename, filename))
-        else:
-            formatted_results.append((filename, filename))
+        formatted_results.append((filename, filename))
     
     return formatted_results
 
@@ -168,10 +146,10 @@ def create_vibecheck_results_ui():
 
     with gr.Blocks(theme='davehornik/Tealy', css=custom_css) as demo:
         gr.HTML("""
-        <div class="header-text">
-            <h1><span class="header-emoji">✨</span> VibeCheck Results Viewer <span class="header-emoji">✨</span></h1>
-            <p>Explore previously saved VibeCheck analysis results</p>
-        </div>
+            <div class="header-text">
+                <h1><span class="header-emoji">✨</span> LLM VibeChecks <span class="header-emoji">✨</span></h1>
+                <p>Visualize the Vibes of model pairs using <a href="https://arxiv.org/abs/2410.12851" target="_blank" style="text-decoration: underline; color: inherit;">VibeCheck system</a> (more to come)</p>
+            </div>
         """)
 
         gr.HTML("""
@@ -295,7 +273,6 @@ def create_vibecheck_results_ui():
             # Summarize
             summary_text = (
                 "## Loaded VibeCheck Results\n"
-                + f"Wandb Run URL: {results.get('wandb_run_url', 'N/A')}\n\n"
                 + model_info
                 + f"**Found {len(results['vibe_df']['vibe'].unique())} vibe(s)**\n"
                 + "\n".join(
@@ -347,8 +324,7 @@ def create_vibecheck_results_ui():
         # Add a footer
         gr.HTML("""
         <div class="footer">
-            <p>VibeCheck Results Viewer</p>
-            <p>© 2023 VibeCheck Team</p>
+            <p>© just some folks who like vibes</p>
         </div>
         """)