synapse-sdk 1.0.0a40__tar.gz → 1.0.0a41__tar.gz

{synapse_sdk-1.0.0a40/synapse_sdk.egg-info → synapse_sdk-1.0.0a41}/PKG-INFO

@@ -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 → synapse_sdk-1.0.0a41}/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-1.0.0a40 → synapse_sdk-1.0.0a41}/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 → synapse_sdk-1.0.0a41/synapse_sdk.egg-info}/PKG-INFO

@@ -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