Update app.py

app.py

@@ -114,6 +114,9 @@ def load_generator(model_name):
 def inference_image(img_path, model_name):
     """Process a single image"""
     try:
+        if img_path is None:
+            return None, "❌ Please upload an image first"
+            
         generator = load_generator(model_name)
         
         print("Processing image:", img_path)
@@ -141,7 +144,7 @@ def inference_image(img_path, model_name):
         return outputs, f"✅ Saved to: {output_path}"
     
     except Exception as error:
-        raise gr.Error(f"Error processing image: {error}")
+        return None, f"❌ Error: {str(error)}"
 
 
 def process_video_frame_by_frame(video_path, model_name, task_id):
@@ -292,7 +295,7 @@ def clear_history():
     """Clear all history"""
     if os.path.exists(HISTORY_FILE):
         os.remove(HISTORY_FILE)
-    return "✅ History cleared!"
+    return "✅ History cleared!", get_history_display()
 
 
 if __name__ == '__main__':
@@ -301,20 +304,20 @@ if __name__ == '__main__':
     worker_thread = threading.Thread(target=video_queue_worker, daemon=True)
     worker_thread.start()
     
-    MARKDOWN = \
-    """
-    ## <p style='text-align: center'> APISR: Anime Production Inspired Real-World Anime Super-Resolution (CVPR 2024) </p>
-    
-    [GitHub](https://github.com/Kiteretsu77/APISR) | [Paper](https://arxiv.org/abs/2403.01598)
+    MARKDOWN = """
+# APISR: Anime Production Inspired Real-World Anime Super-Resolution (CVPR 2024)
 
-    APISR aims at restoring and enhancing low-quality low-resolution **anime** images and video sources with various degradations from real-world scenarios.
-    
-    ### ⚠️ Note: Images with short side > 720px will be downsampled to 720px (e.g., 1920x1080 → 1280x720)
-    ### 📹 New: Video processing runs in background queue - you can close the browser and it continues!
-    """
+[GitHub](https://github.com/Kiteretsu77/APISR) | [Paper](https://arxiv.org/abs/2403.01598)
+
+APISR aims at restoring and enhancing low-quality low-resolution **anime** images and video sources with various degradations from real-world scenarios.
 
-    block = gr.Blocks().queue(max_size=10)
-    with block:
+### ⚠️ Note: Images with short side > 720px will be downsampled to 720px (e.g., 1920x1080 → 1280x720)
+### 📹 New: Video processing runs in background queue - you can close the browser and it continues!
+"""
+
+    # Create Gradio interface with Gradio 5.x/6.x syntax
+    with gr.Blocks(title="APISR - Anime Super Resolution", theme=gr.themes.Soft()) as demo:
+        
         gr.Markdown(MARKDOWN)
         
         with gr.Tabs():
@@ -324,7 +327,7 @@ if __name__ == '__main__':
                     with gr.Column(scale=2):
                         input_image = gr.Image(type="filepath", label="Input Image")
                         image_model = gr.Dropdown(
-                            ["2xRRDB", "4xRRDB", "4xGRL", "4xDAT"],
+                            choices=["2xRRDB", "4xRRDB", "4xGRL", "4xDAT"],
                             value="4xGRL",
                             label="Model"
                         )
@@ -336,17 +339,17 @@ if __name__ == '__main__':
                 
                 with gr.Row():
                     gr.Examples(
-                        [
+                        examples=[
                             ["__assets__/lr_inputs/image-00277.png"],
                             ["__assets__/lr_inputs/image-00542.png"],
                             ["__assets__/lr_inputs/41.png"],
                             ["__assets__/lr_inputs/f91.jpg"],
                         ],
-                        [input_image],
+                        inputs=[input_image],
                     )
                 
                 image_btn.click(
-                    inference_image, 
+                    fn=inference_image, 
                     inputs=[input_image, image_model], 
                     outputs=[output_image, image_status]
                 )
@@ -362,7 +365,7 @@ if __name__ == '__main__':
                     with gr.Column():
                         input_video = gr.Video(label="Input Video")
                         video_model = gr.Dropdown(
-                            ["2xRRDB", "4xRRDB", "4xGRL", "4xDAT"],
+                            choices=["2xRRDB", "4xRRDB", "4xGRL", "4xDAT"],
                             value="4xGRL",
                             label="Model"
                         )
@@ -371,19 +374,22 @@ if __name__ == '__main__':
                     
                     with gr.Column():
                         gr.Markdown("### 📊 Queue Monitor")
-                        queue_status = gr.Textbox(label="Queue Status", lines=15)
+                        queue_status = gr.Textbox(label="Queue Status", lines=15, interactive=False)
                         refresh_btn = gr.Button("🔄 Refresh Status")
                 
                 video_btn.click(
-                    submit_video,
+                    fn=submit_video,
                     inputs=[input_video, video_model],
                     outputs=[input_video, video_status]
                 )
                 
                 refresh_btn.click(
-                    get_queue_status,
+                    fn=get_queue_status,
                     outputs=[queue_status]
                 )
+                
+                # Auto-refresh every 5 seconds
+                demo.load(fn=get_queue_status, outputs=[queue_status], every=5)
             
             # Tab 3: History
             with gr.Tab("📜 History"):
@@ -393,20 +399,26 @@ if __name__ == '__main__':
                     refresh_history_btn = gr.Button("🔄 Refresh History")
                     clear_history_btn = gr.Button("🗑️ Clear History", variant="stop")
                 
-                history_display = gr.Textbox(label="History", lines=20)
-                clear_status = gr.Textbox(label="Status", lines=1)
+                history_display = gr.Textbox(label="History", lines=20, interactive=False)
+                clear_status = gr.Textbox(label="Status", lines=1, visible=True)
                 
                 refresh_history_btn.click(
-                    get_history_display,
+                    fn=get_history_display,
                     outputs=[history_display]
                 )
                 
                 clear_history_btn.click(
-                    clear_history,
-                    outputs=[clear_status]
+                    fn=clear_history,
+                    outputs=[clear_status, history_display]
                 )
                 
-                # Auto-load history on tab open
-                block.load(get_history_display, outputs=[history_display])
-
-    block.launch(server_name="0.0.0.0", server_port=7860)
+                # Auto-load history on page load
+                demo.load(fn=get_history_display, outputs=[history_display])
+
+    # Launch the app
+    demo.queue(max_size=20)
+    demo.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        share=False
+    )