Update app.py

app.py

@@ -1,12 +1,11 @@
 import os
 import requests
 import gradio as gr
+from datetime import datetime
 
 # API Configuration
 API_KEY = os.getenv('StableCogKey')
 API_HOST = 'https://api.stablecog.com'
-MODELS_ENDPOINT = '/v1/image/generation/models'
-URL = f'{API_HOST}{MODELS_ENDPOINT}'
 
 headers = {
     'Authorization': f'Bearer {API_KEY}',
@@ -16,47 +15,178 @@ headers = {
 def get_models():
     """Fetch and display available models"""
     try:
-        response = requests.get(URL, headers=headers, timeout=10)
+        url = f'{API_HOST}/v1/image/generation/models'
+        response = requests.get(url, headers=headers, timeout=10)
         
         if response.status_code == 200:
             data = response.json()
             models = data.get('models', [])
             
             # Format display
-            display_text = ""
-            for model in models:
+            display_text = f"📊 Found {len(models)} models\n"
+            display_text += f"⏰ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
+            
+            for i, model in enumerate(models, 1):
                 name = model.get('name', 'Unknown')
                 model_type = model.get('type', 'unknown')
                 description = model.get('description', 'No description')
                 
-                display_text += f"🔹 {name} ({model_type})\n"
+                display_text += f"{i}. 🔹 **{name}**\n"
                 display_text += f"   📝 {description}\n"
-                display_text += f"   📊 Public: {model.get('is_public', False)}\n"
-                display_text += f"   ⭐ Default: {model.get('is_default', False)}\n"
-                display_text += f"   👥 Community: {model.get('is_community', False)}\n"
+                display_text += f"   🏷️ Type: {model_type}\n"
+                display_text += f"   🌍 Public: {'✅' if model.get('is_public') else '❌'}\n"
+                display_text += f"   ⭐ Default: {'✅' if model.get('is_default') else '❌'}\n"
+                display_text += f"   👥 Community: {'✅' if model.get('is_community') else '❌'}\n"
                 display_text += "─" * 40 + "\n"
             
-            total = len(models)
-            display_text = f"📊 Found {total} models:\n\n" + display_text
-            
             return display_text, str(data)
             
         else:
-            return f"❌ Error: {response.status_code}", f"Error: {response.text}"
+            return f"❌ Error {response.status_code}", f"Error: {response.text}"
             
     except Exception as e:
-        return f"❌ Connection error: {str(e)}", "No data"
+        return f"❌ Error: {str(e)}", "No data"
+
+def get_outputs():
+    """Fetch and display recent outputs with images"""
+    try:
+        url = f'{API_HOST}/v1/image/generation/outputs'
+        response = requests.get(url, headers=headers, timeout=10)
+        
+        if response.status_code == 200:
+            data = response.json()
+            outputs = data.get('outputs', [])
+            total = data.get('total_count', 0)
+            next_cursor = data.get('next', 'None')
+            
+            # Prepare display components
+            image_gallery = []
+            display_text = ""
+            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 '🤍'
+                    
+                    # 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')
+                    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>
+                        </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>"
+            
+            gallery_html += "</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}"
+            
+    except Exception as e:
+        error_text = f"❌ Error: {str(e)}"
+        return error_text, "<div style='color: red; padding: 20px;'>" + error_text + "</div>", "No data"
 
 # Create Gradio interface
-with gr.Blocks(title="StableCog Models") as demo:
-    gr.Markdown("# 🤖 StableCog Models Checker")
+with gr.Blocks(title="StableCog Dashboard") as demo:
+    gr.Markdown("# 🎯 StableCog Dashboard")
     
-    with gr.Row():
-        models_display = gr.Textbox(label="Models", lines=20)
-        raw_output = gr.Code(label="Raw JSON", language="json", lines=20)
-    
-    check_btn = gr.Button("🔄 Check Models", variant="primary")
-    check_btn.click(get_models, outputs=[models_display, raw_output])
+    with gr.Tabs():
+        with gr.Tab("🤖 Models"):
+            with gr.Row():
+                models_display = gr.Textbox(label="Available Models", lines=25)
+                models_raw = gr.Code(label="Raw JSON", language="json", lines=25)
+            
+            check_models_btn = gr.Button("🔄 Refresh Models", variant="primary")
+            check_models_btn.click(get_models, outputs=[models_display, models_raw])
+        
+        with gr.Tab("🖼️ Outputs"):
+            with gr.Row():
+                with gr.Column(scale=2):
+                    outputs_display = gr.Textbox(label="Output Details", lines=25)
+                with gr.Column(scale=3):
+                    outputs_gallery = gr.HTML(label="Image Gallery")
+            
+            with gr.Row():
+                outputs_raw = gr.Code(label="Raw JSON", language="json", lines=15)
+            
+            check_outputs_btn = gr.Button("🔄 Refresh Outputs", variant="primary")
+            check_outputs_btn.click(get_outputs, outputs=[outputs_display, outputs_gallery, outputs_raw])
 
 if __name__ == "__main__":
     demo.launch()