Update app.py

app.py

@@ -59,108 +59,106 @@ def get_outputs():
             total = data.get('total_count', 0)
             next_cursor = data.get('next', 'None')
             
-            # Prepare display components
-            image_gallery = []
-            display_text = ""
+            # Prepare display
+            display_text = f"📊 Total outputs: {total}\n"
+            display_text += f"📋 Showing: {len(outputs)} outputs\n"
+            display_text += f"⏭️ Next cursor: {next_cursor}\n"
+            display_text += f"⏰ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
+            
+            # Create gallery HTML
             gallery_html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 15px;'>"
             
-            if outputs:
-                display_text += f"📊 Total outputs: {total}\n"
-                display_text += f"📋 Showing: {len(outputs)} outputs\n"
-                display_text += f"⏭️ Next cursor: {next_cursor}\n"
-                display_text += f"⏰ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
+            for i, output in enumerate(outputs, 1):
+                output_id = output.get('id', 'N/A')
+                short_id = output_id[:8] + '...' if len(output_id) > 8 else output_id
+                created_at = output.get('created_at', 'N/A')
+                status = output.get('status', 'unknown')
+                model_name = output.get('model_name', 'Unknown')
+                
+                # Gallery status with emojis
+                gallery_status = output.get('gallery_status', 'not_submitted')
+                gallery_emoji = {
+                    'not_submitted': '🔒',
+                    'submitted': '📤',
+                    'approved': '✅',
+                    'rejected': '❌'
+                }.get(gallery_status, '❓')
+                
+                # Favorites
+                is_favorited = output.get('is_favorited', False)
+                favorite_emoji = '❤️' if is_favorited else '🤍'
                 
-                for i, output in enumerate(outputs, 1):
-                    output_id = output.get('id', 'N/A')
-                    short_id = output_id[:8] + '...' if len(output_id) > 8 else output_id
-                    created_at = output.get('created_at', 'N/A')
-                    status = output.get('status', 'unknown')
-                    model_name = output.get('model_name', 'Unknown')
-                    
-                    # Gallery status with emojis
-                    gallery_status = output.get('gallery_status', 'not_submitted')
-                    gallery_emoji = {
-                        'not_submitted': '🔒',
-                        'submitted': '📤',
-                        'approved': '✅',
-                        'rejected': '❌'
-                    }.get(gallery_status, '❓')
-                    
-                    # Favorites
-                    is_favorited = output.get('is_favorited', False)
-                    favorite_emoji = '❤️' if is_favorited else '🤍'
-                    
-                    # Auto-submitted
-                    was_auto_submitted = output.get('was_auto_submitted', False)
-                    auto_submit_emoji = '🤖' if was_auto_submitted else '👤'
-                    
-                    # Format timestamp
-                    if created_at != 'N/A':
-                        try:
-                            dt = datetime.fromisoformat(created_at.replace('Z', '+00:00'))
-                            created_at = dt.strftime('%Y-%m-%d %H:%M')
-                        except:
-                            pass
-                    
-                    display_text += f"{i}. 🔹 **Output {short_id}**\n"
-                    display_text += f"   🕒 Created: {created_at}\n"
-                    display_text += f"   📊 Status: {status}\n"
-                    display_text += f"   🤖 Model: {model_name}\n"
-                    display_text += f"   🖼️ Gallery: {gallery_emoji} {gallery_status}\n"
-                    display_text += f"   {favorite_emoji} Favorite\n"
-                    display_text += f"   {auto_submit_emoji} {'Auto-submitted' if was_auto_submitted else 'Manual'}\n"
-                    
-                    # Get image URLs
-                    image_url = output.get('image_url')
+                # Format timestamp
+                if created_at != 'N/A':
+                    try:
+                        dt = datetime.fromisoformat(created_at.replace('Z', '+00:00'))
+                        created_date = dt.strftime('%Y-%m-%d')
+                        created_time = dt.strftime('%H:%M')
+                    except:
+                        created_date = created_at
+                        created_time = ''
+                else:
+                    created_date = 'Unknown'
+                    created_time = ''
+                
+                # Text display
+                display_text += f"{i}. 🔹 **Output {short_id}**\n"
+                display_text += f"   🕒 Created: {created_at}\n"
+                display_text += f"   📊 Status: {status}\n"
+                display_text += f"   🤖 Model: {model_name}\n"
+                display_text += f"   🖼️ Gallery: {gallery_emoji} {gallery_status}\n"
+                display_text += f"   {favorite_emoji} Favorite\n"
+                
+                # Get image URL
+                image_url = output.get('image_url')
+                if not image_url:
+                    # Try to get from image_urls array
                     image_urls = output.get('image_urls', [])
-                    
-                    # Use image_url if available, otherwise check image_urls array
-                    image_to_show = image_url or (image_urls[0] if image_urls else None)
-                    
-                    if image_to_show:
-                        # Add to gallery
-                        gallery_html += f"""
-                        <div style='border-radius: 10px; overflow: hidden; background: rgba(255,255,255,0.1); padding: 10px;'>
-                            <img src='{image_to_show}' style='width: 100%; height: 200px; object-fit: cover; border-radius: 8px;'>
-                            <div style='margin-top: 8px; font-size: 12px;'>
-                                <div>📅 {created_at.split()[0] if ' ' in created_at else created_at}</div>
-                                <div>🤖 {model_name[:15]}{'...' if len(model_name) > 15 else ''}</div>
-                                <div>{gallery_emoji} {gallery_status}</div>
-                                <div>{favorite_emoji}</div>
-                            </div>
+                    if image_urls and isinstance(image_urls, list) and len(image_urls) > 0:
+                        image_url = image_urls[0]
+                
+                if image_url:
+                    # Add to gallery
+                    gallery_html += f"""
+                    <div style='border-radius: 10px; overflow: hidden; background: rgba(255,255,255,0.1); padding: 10px;'>
+                        <img src='{image_url}' style='width: 100%; height: 200px; object-fit: cover; border-radius: 8px;'>
+                        <div style='margin-top: 8px; font-size: 12px;'>
+                            <div>📅 {created_date}</div>
+                            <div>🤖 {model_name[:15]}{'...' if len(model_name) > 15 else ''}</div>
+                            <div>{gallery_emoji} {gallery_status}</div>
+                            <div>{favorite_emoji}</div>
                         </div>
-                        """
-                        
-                        display_text += f"   🖼️ Image: {image_to_show[:50]}...\n"
-                    else:
-                        display_text += f"   🖼️ No image available\n"
-                    
-                    # Show generation details if available
-                    generation = output.get('generation', {})
-                    if generation:
-                        prompt = generation.get('prompt', 'No prompt')
-                        width = generation.get('width', '?')
-                        height = generation.get('height', '?')
-                        
-                        display_text += f"   📝 Prompt: {prompt[:50]}...\n"
-                        display_text += f"   📐 Size: {width}x{height}\n"
-                    
-                    display_text += "─" * 40 + "\n"
-            else:
-                display_text = "📭 No outputs found. Generate some images first!"
-                gallery_html += "<div style='text-align: center; padding: 40px;'>No images found</div>"
+                    </div>
+                    """
+                    display_text += f"   🖼️ Image URL: {image_url[:50]}...\n"
+                else:
+                    display_text += f"   🖼️ No image available\n"
+                
+                # Show generation details if available
+                generation = output.get('generation', {})
+                if generation:
+                    prompt = generation.get('prompt', 'No prompt')
+                    if len(prompt) > 50:
+                        prompt = prompt[:50] + '...'
+                    display_text += f"   📝 Prompt: {prompt}\n"
+                
+                display_text += "─" * 40 + "\n"
             
             gallery_html += "</div>"
             
+            if not outputs:
+                display_text = "📭 No outputs found. Generate some images first!"
+                gallery_html = "<div style='text-align: center; padding: 40px; color: #888;'>No images found</div>"
+            
             return display_text, gallery_html, str(data)
             
         else:
-            error_text = f"❌ Error {response.status_code}"
-            return error_text, "<div style='color: red; padding: 20px;'>" + error_text + "</div>", f"Error: {response.text}"
+            error_msg = f"❌ Error {response.status_code}"
+            return error_msg, f"<div style='color: red; padding: 20px;'>{error_msg}</div>", f"Error: {response.text}"
             
     except Exception as e:
-        error_text = f"❌ Error: {str(e)}"
-        return error_text, "<div style='color: red; padding: 20px;'>" + error_text + "</div>", "No data"
+        error_msg = f"❌ Error: {str(e)}"
+        return error_msg, f"<div style='color: red; padding: 20px;'>{error_msg}</div>", "No data"
 
 # Create Gradio interface
 with gr.Blocks(title="StableCog Dashboard") as demo:
@@ -177,9 +175,9 @@ with gr.Blocks(title="StableCog Dashboard") as demo:
         
         with gr.Tab("🖼️ Outputs"):
             with gr.Row():
-                with gr.Column(scale=2):
+                with gr.Column(scale=1):
                     outputs_display = gr.Textbox(label="Output Details", lines=25)
-                with gr.Column(scale=3):
+                with gr.Column(scale=2):
                     outputs_gallery = gr.HTML(label="Image Gallery")
             
             with gr.Row():