Google colab対応

ウィジットの名称をmyappに変更

app.py

@@ -3,29 +3,35 @@
     SaliencyMapDemo
 """
 import argparse
-from datetime import datetime
 
-from utils import get_package_version, Stopwatch
-from reporter import get_current_reporter
+import src.utils as utils
+from src.reporter import get_current_reporter
+        
+PROGRAM_NAME = 'SaliencyMapDemo'
+__version__ = utils.get_package_version()
 
 def main():
-	"""
-		エントリーポイント
-		1, コマンドライン引数の解析を行います
-		2, アプリを起動します。
-	"""
-	log = get_current_reporter()
-	log.info("#アプリ起動中")
-	watch = Stopwatch.startNew()
-	
-	import app
+    """
+        エントリーポイント
+        1, コマンドライン引数の解析を行います
+        2, アプリを起動します。
+    """
+    log = get_current_reporter()
+    log.info("#アプリ起動中")
+    watch = utils.Stopwatch.startNew()
+    
+    from src.myapp import runApp
 
-	parser = argparse.ArgumentParser(prog=app.PROGRAM_NAME, description="SaliencyMapDemo")
-	parser.add_argument('--server_port', type=int, default=9999, help="Gradio server port")
-	parser.add_argument('--max_file_size', type=int, default=20 * 1024 * 1024, help="Gradio max file size")
-	parser.add_argument('--version', action='version', version=f'%(prog)s {get_package_version()}')
-	
-	app.run(parser.parse_args(), watch)
+    parser = argparse.ArgumentParser(prog=PROGRAM_NAME, description="SaliencyMapDemo")
+    parser.add_argument('--server_port', type=int, default=9999, help="Gradio server port")
+    parser.add_argument('--max_file_size', type=int, default=20 * 1024 * 1024, help="Gradio max file size")
+    parser.add_argument('--inbrowser', type=bool, default=True, help="Gradio inbrowser")
+    parser.add_argument('--share', type=bool, default=False, help="Gradio share")
+    parser.add_argument('--version', action='version', version=f'%(prog)s {__version__}')
+    
+    args = parser.parse_args()
+    
+    runApp(args, watch)
 
 if __name__ == "__main__":
     main()

src/{app.py → myapp.py}

@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
-"""
-    SaliencyMapDemo
-"""
+from typing import Literal
+
 import argparse
 from datetime import datetime
 import sys
@@ -9,14 +8,14 @@ import sys
 import gradio as gr
 import numpy as np
 
-import utils
-from saliency import SaliencyMap, convertColorMap
-from reporter import get_current_reporter
-        
+import src.utils as utils
+from src.saliency import SaliencyMap, convertColorMap
+from src.reporter import get_current_reporter
+
 PROGRAM_NAME = 'SaliencyMapDemo'
 __version__ = utils.get_package_version()
 log = get_current_reporter()
-
+    
 def jetTab_Selected(image: np.ndarray):
     #print(f"{datetime.now()}#jet")
     saliency = SaliencyMap("SpectralResidual")
@@ -30,7 +29,7 @@ def hotTab_Selected(image: np.ndarray):
     #print(f"{datetime.now()}#hot")
     saliency = SaliencyMap("SpectralResidual")
     success, saliencyMap = saliency.computeSaliency(image)
-    retval = convertColorMap(image, saliencyMap, "hot")
+    retval = convertColorMap(image, saliencyMap, "turbo")
     #print(f"{datetime.now()}#hot")
     
     return retval
@@ -60,13 +59,14 @@ def submit_Clicked(image: np.ndarray, algorithm: str):
     jet = convertColorMap(image, saliencyMap, "jet")
     #jet = None
     log.info(f"#hot")
-    hot = convertColorMap(image, saliencyMap, "hot")
+    hot = convertColorMap(image, saliencyMap, "turbo")
     
     saliency = None
     log.info(f"#submit_Clicked End{watch.stop():.3f}")
+    
     return jet, hot
 
-def run(args: argparse.Namespace, watch: utils.Stopwatch) -> None:
+def runApp(args: argparse.Namespace, watch: utils.Stopwatch) -> None:
     """
     アプリの画面を作成し、Gradioサービスを起動します。
 
@@ -123,11 +123,12 @@ def run(args: argparse.Namespace, watch: utils.Stopwatch) -> None:
         demo.queue(default_concurrency_limit=5)
         
         log.info(f"#アプリ起動完了({watch.stop():.3f}s)")
-        
         # https://www.gradio.app/docs/gradio/blocks#blocks-launch
+        
         demo.launch(
             max_file_size=args.max_file_size,
             server_port=args.server_port,
-            inbrowser=True,
-            share=False,
+            inbrowser=args.inbrowser,
+            share=args.share,
         )
+