add cost

app.py

@@ -94,7 +94,7 @@ def get_base_filename(filepath):
 
 # Define the Gradio interface
 def create_demo():
-    # --- Reintroduce CSS ---
+    # --- Updated CSS with model info styling ---
     custom_css = """
     /* Container for the image component (#current-image-display is the elem_id of gr.Image) */
         #current-image-display {
@@ -115,6 +115,15 @@ def create_demo():
             height: auto;           /* Use natural height unless constrained by max-height */
             display: block;         /* Ensure image behaves predictably in flex */
         }
+        
+        /* Custom style for model info display */
+        #model-info-display {
+            font-size: 0.85rem;     /* Small font size */
+            color: #666;            /* Subtle color */
+            margin-top: 0.5rem;     /* Small top margin */
+            margin-bottom: 1rem;    /* Bottom margin before next element */
+            padding-left: 0.5rem;   /* Slight indentation */
+        }
     """
     # --- Pass css to gr.Blocks ---
     with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
@@ -154,7 +163,7 @@ def create_demo():
         ]
         
         # Calculate all models once
-        all_models = preferred_models + additional_models
+        all_models_list = preferred_models + additional_models
         
         # Default model value
         default_model = "google/gemini-2.0-flash-001"
@@ -195,12 +204,20 @@ def create_demo():
                         label="Content Type",
                         value="Museum Object"
                     )
-                    
-                    # Display current model info
-                    model_info = gr.Markdown(
-                        f"""**Current Model**: Gemini 2.0 Flash (default)  
-                        **Cost per 100 Images**: {MODEL_PRICING[default_model]}"""
-                    )
+                
+                # Find the default model's display name
+                default_model_name = "Unknown Model"
+                for name, value in preferred_models:
+                    if value == default_model:
+                        default_model_name = name
+                        break
+                
+                # Model info outside the accordion
+                model_info = gr.Markdown(
+                    f"""**Current Model**: {default_model_name}  
+                    **Cost per 100 Images**: {MODEL_PRICING[default_model]}""",
+                    elem_id="model-info-display"
+                )
                 
                 gr.Markdown("### Uploaded Images")
                 input_gallery = gr.Gallery(
@@ -239,11 +256,11 @@ def create_demo():
                 all_images = gr.State([])
                 all_results = gr.State([])
         
-        # Handle checkbox change to update model dropdown
+        # Handle checkbox change to update model dropdown - modern version
         def toggle_models(show_all, current_model):
             # Make a fresh copy of the models lists to avoid any reference issues
             preferred_choices = list(preferred_models)
-            all_choices = list(all_models)
+            all_choices = list(all_models_list)
             
             if show_all:
                 # When showing all models, use the fresh copy of all models
@@ -258,23 +275,15 @@ def create_demo():
                 else:
                     # Reset to default model if current model is not in preferred models
                     return gr.Dropdown(choices=preferred_choices, value=default_model)
-        
-        # Connect checkbox to toggle model choices - keep this the same
-        show_all_models.change(
-            fn=toggle_models,
-            inputs=[show_all_models, model_choice],
-            outputs=[model_choice]
-        )
 
         # Update model info when model selection changes
         def update_model_info(model_value):
             # Find display name
-            for name, value in all_models:
+            model_name = "Unknown Model"
+            for name, value in all_models_list:
                 if value == model_value:
                     model_name = name
                     break
-            else:
-                model_name = "Unknown Model"
             
             # Get cost
             cost = MODEL_PRICING.get(model_value, "Unknown")
@@ -290,6 +299,13 @@ def create_demo():
             outputs=[model_choice]
         )
         
+        # Connect model selection to update info
+        model_choice.change(
+            fn=update_model_info,
+            inputs=[model_choice],
+            outputs=[model_info]
+        )
+        
         # Handle file uploads
         def handle_upload(files, current_paths, current_filenames):
             file_paths = []