synapse-sdk 1.0.0a40__py3-none-any.whl → 1.0.0a41__py3-none-any.whl

synapse_sdk/plugins/categories/neural_net/actions/gradio.py

@@ -22,8 +22,10 @@ class GradioAction(Action):
         return dir
 
     @property
-    def _requirements(self):
-        return self.config.get('requirements', ['gradio>=5'])
+    def requirements_file(self):
+        requirements_file = self.working_directory / 'requirements.txt'
+        if requirements_file.exists():
+            return requirements_file
 
     @property
     def tag(self):
@@ -40,7 +42,7 @@ class GradioAction(Action):
         try:
             # Write Dockerfile and requirements.txt
             path_dockerfile = self.write_dockerfile_template()
-            self.write_requirements(path_dockerfile.parent / 'requirements.txt')
+            self.check_requirements()
 
             # Build docker image
             self.build_docker_image(path_dockerfile)
@@ -59,11 +61,15 @@ class GradioAction(Action):
         dockerfile_path = self.working_directory / 'Dockerfile'
 
         with open(dockerfile_path, 'w') as f:
-            f.write("""FROM python:3.10
+            f.write("""FROM python:3.12-slim
 WORKDIR /home/user/app
 
 RUN pip install --no-cache-dir pip -U && \\
     pip install --no-cache-dir uvicorn
+    
+RUN apt-get update && \\
+    apt-get install -y git nmap ffmpeg libsm6 libxext6 libgl1-mesa-glx && \\
+    rm -rf /var/lib/apt/lists/*
 
 RUN apt-get update && \\
     apt-get install -y curl && \\
@@ -82,22 +88,19 @@ CMD ["python", "app.py"]
 """)
         return dockerfile_path
 
-    def write_requirements(self, path):
-        with open(path, 'w') as f:
-            f.write('\n'.join(self._requirements))
+    def check_requirements(self):
+        default_packages = ['gradio', 'synapse-sdk', 'python-nmap']
+        if self.requirements_file is None:
+            with open(self.working_directory / 'requirements.txt', 'a') as f:
+                f.write('\n' + '\n'.join(default_packages))
+        else:
+            with open(self.requirements_file, 'a') as f:
+                f.write('\n' + '\n'.join(default_packages[1:]))
 
     def build_docker_image(self, path_dockerfile):
         self.run.log('deploy', 'Start building docker image')
         result = subprocess.run(
-            [
-                'docker',
-                'build',
-                '-t',
-                self.tag,
-                '-f',
-                str(path_dockerfile),
-                '.',
-            ],
+            ['docker', 'build', '-t', self.tag, '-f', str(path_dockerfile), '.'],
             cwd=self.working_directory,
             check=True,
         )
@@ -113,19 +116,30 @@ CMD ["python", "app.py"]
             subprocess.run(['docker', 'rm', self.tag], check=True)
 
         # Run docker image
-        self.run.log('deploy', 'Starting docker container')
+        command = [
+            'docker',
+            'run',
+            '-d',
+            '--name',
+            self.tag,
+            '-p',
+            f'{self.deploy_port}:7860',
+            '-p',
+            '8991-8999:8991-8999',
+            '--add-host',
+            'host.docker.internal:host-gateway',
+            '-e',
+            'GRADIO_SERVER_NAME=0.0.0.0',
+        ]
+
+        # extend synapse env vars
+        for key, value in self.envs.items():
+            command.extend(['-e', f'{key}={value}'])
+        command.append(self.tag)
+
+        self.run.log('deploy', f'Starting docker container with command: {" ".join(command)}')
+
         subprocess.run(
-            [
-                'docker',
-                'run',
-                '-d',
-                '--name',
-                self.tag,
-                '-p',
-                f'{self.deploy_port}:7860',
-                '-p',
-                '8991-8999:8991-8999',
-                self.tag,
-            ],
+            command,
             check=True,
         )

synapse_sdk/utils/http.py

@@ -38,7 +38,7 @@ class SingleFileHttpServer(SimpleHTTPRequestHandler):
 
 
 @contextmanager
-def temp_file_server(image=None, file_path=None, format='JPEG', host='localhost', port=None, content_type=None):
+def temp_file_server(image=None, file_path=None, format='JPEG', host='0.0.0.0', port=None, content_type=None):
     """
     Context manager that serves a file temporarily via HTTP.
 
@@ -46,7 +46,7 @@ def temp_file_server(image=None, file_path=None, format='JPEG', host='localhost'
         image: A PIL Image object to serve (optional)
         file_path: Path to an existing file to serve (optional - used if image not provided)
         format: Image format when saving a PIL Image (default: "JPEG")
-        host: Host to serve on (default: "localhost")
+        host: Host to serve on (default: "0.0.0.0")
         port: Port to serve on (default: auto-selected free port)
         content_type: Content type header (default: auto-detected based on format)
 
@@ -92,7 +92,7 @@ def temp_file_server(image=None, file_path=None, format='JPEG', host='localhost'
 
         time.sleep(0.1)
 
-        url = f'http://{host}:{port}'
+        url = f'http://localhost:{port}'
 
         try:
             yield url

{synapse_sdk-1.0.0a40.dist-info → synapse_sdk-1.0.0a41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a40
+Version: 1.0.0a41
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a40.dist-info → synapse_sdk-1.0.0a41.dist-info}/RECORD

@@ -68,7 +68,7 @@ synapse_sdk/plugins/categories/export/templates/plugin/export.py,sha256=39XLGo8u
 synapse_sdk/plugins/categories/neural_net/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/neural_net/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/neural_net/actions/deployment.py,sha256=y2LrS-pwazqRI5O0q1NUy45NQYsBj6ykbrXnDMs_fqE,1987
-synapse_sdk/plugins/categories/neural_net/actions/gradio.py,sha256=pcrMr3pmNJmVjS2wRvyGmCvIoxYKSQJAheXqJBNRfm4,3790
+synapse_sdk/plugins/categories/neural_net/actions/gradio.py,sha256=246hZyiRb8z2JtohbEOFW2QN56vli_3LciexEmQYX7w,4444
 synapse_sdk/plugins/categories/neural_net/actions/inference.py,sha256=0a655ELqNVjPFZTJDiw4EUdcMCPGveUEKyoYqpwMFBU,1019
 synapse_sdk/plugins/categories/neural_net/actions/test.py,sha256=JY25eg-Fo6WbgtMkGoo_qNqoaZkp3AQNEypJmeGzEog,320
 synapse_sdk/plugins/categories/neural_net/actions/train.py,sha256=kve6iTCg2kUeavMQTR2JFuoYDu-QWZFFlB58ZICQtdM,5406
@@ -120,7 +120,7 @@ synapse_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 synapse_sdk/utils/dataset.py,sha256=zWTzFmv589izFr62BDuApi3r5FpTsdm-5AmriC0AEdM,1865
 synapse_sdk/utils/debug.py,sha256=F7JlUwYjTFZAMRbBqKm6hxOIz-_IXYA8lBInOS4jbS4,100
 synapse_sdk/utils/file.py,sha256=wWBQAx0cB5a-fjfRMeJV-KjBil1ZyKRz-vXno3xBSoo,6834
-synapse_sdk/utils/http.py,sha256=GqYONHfovwBmP4p3ZczVCNvn4oApx2QzfzGIEiICqJo,3770
+synapse_sdk/utils/http.py,sha256=VyehrKHPnCrXXSYLmqZXTAOYW1-LhrWyWnnLr8MVKNA,3769
 synapse_sdk/utils/module_loading.py,sha256=chHpU-BZjtYaTBD_q0T7LcKWtqKvYBS4L0lPlKkoMQ8,1020
 synapse_sdk/utils/network.py,sha256=WI8qn6KlKpHdMi45V57ofKJB8zusJrbQsxT74LwVfsY,1000
 synapse_sdk/utils/string.py,sha256=rEwuZ9SAaZLcQ8TYiwNKr1h2u4CfnrQx7SUL8NWmChg,216
@@ -134,9 +134,9 @@ synapse_sdk/utils/storage/providers/__init__.py,sha256=x7RGwZryT2FpVxS7fGWryRVpq
 synapse_sdk/utils/storage/providers/gcp.py,sha256=i2BQCu1Kej1If9SuNr2_lEyTcr5M_ncGITZrL0u5wEA,363
 synapse_sdk/utils/storage/providers/s3.py,sha256=W94rQvhGRXti3R4mYP7gmU5pcyCQpGFIBLvxxqLVdRM,2231
 synapse_sdk/utils/storage/providers/sftp.py,sha256=_8s9hf0JXIO21gvm-JVS00FbLsbtvly4c-ETLRax68A,1426
-synapse_sdk-1.0.0a40.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0a40.dist-info/METADATA,sha256=P24axYuG6BpxcsvfyKnMuZh6NpNBQCaN3gcRnggwxDg,1203
-synapse_sdk-1.0.0a40.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-synapse_sdk-1.0.0a40.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0a40.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0a40.dist-info/RECORD,,
+synapse_sdk-1.0.0a41.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0a41.dist-info/METADATA,sha256=Txz9DmdCWS8KzIWa2XEpLM0YYOzlCgcTQUXtDYxCw0c,1203
+synapse_sdk-1.0.0a41.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+synapse_sdk-1.0.0a41.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0a41.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0a41.dist-info/RECORD,,