refactor 💻

backend/src/__init__.py

@@ -1,4 +1,3 @@
-import resources
-from .example.examples import *
+import resource
 import src.demo 
 import src.index

backend/src/demo.py

@@ -5,9 +5,9 @@ from resources import register, tabularGradio
 def Hello_World(name):
         return f"Hello {name}, and welcome to Gradio Flow 🤗" 
 
-@register(["number"], ["number"], examples=[[1,1]])
+@register(["number", "number"], ["number"], examples=[[1,1]])
 def add(x, y):
     return x + y
 
 if __name__ == "__main__":
-    tabularGradio([Hello_World()], ["Hello World"])
+    tabularGradio([Hello_World(), add()], ["Hello World",])

backend/src/example/__init__.py

@@ -0,0 +1 @@
+from .examples import * 

backend/src/index.py

@@ -1,5 +1,4 @@
 from resources import GradioModule, register
-import gradio as gr
 
 @GradioModule
 class Greeting:
@@ -8,12 +7,9 @@ class Greeting:
     def Hello_World(self, name):
         return f"Hello {name}, and welcome to Gradio Flow 🤗" 
 
-@register(["text"], ["text"])
-def Hello_World(name):
-    return f"Hello {name}, and welcome to Gradio Flow 🤗" 
-
+    @register(["number", "number"], ["number"], examples=[[1,1]])
+    def add(self, x, y):
+        return x + y
 
 if __name__ == "__main__":
-    print(Hello_World("Luca"))
-    
-    # Greeting().run(listen=2000) 
+    Greeting().run(listen=2000)

backend/src/resources/dock.py

@@ -1,24 +0,0 @@
-import socket
-import random
-
-class Dock:
-
-    def __init__(self) -> None:
-            self.num_ports = 20
-            self.port_range = (7860, 7880)
-
-    def portConnection(self, port : int):
-            s = socket.socket(
-                socket.AF_INET, socket.SOCK_STREAM)  
-            result = s.connect_ex(("localhost", port))
-            if result == 0: return True
-            return False
-
-    def determinePort(self, max_trial_count=10):
-            trial_count = 0 
-            while trial_count <= max_trial_count:
-                port=random.randint(*self.port_range)
-                if not self.portConnection(port):
-                    return port
-                trial_count += 1
-            raise Exception('Exceeded Max Trial count without finding port')

backend/src/resources/module.py

@@ -32,6 +32,9 @@ DOCKER_LOCAL_HOST = '0.0.0.0'
 DOCKER_PORT = Dock()
 
 def tabularGradio(funcs, names, name="Tabular Temp Name", **kwargs):
+    #print([fn.__name__ for fn in funcs])
+    assert len(funcs) == len(names), f"{bcolor.BOLD}{bcolor.FAIL}🐛 something went wrong!!! The function you appended dose not match the lenght of the names{bcolor.ENDC}"
+    # assert all([fn == "wrap" for fn in funcs]), f"{bcolor().BOLD}{bcolor().FAIL}not all of these are decorated with the right decorator{bcolor().ENDC}"
     port= kwargs["port"] if "port" in kwargs else DOCKER_PORT.determinePort()
 
     try:
@@ -82,7 +85,7 @@ def register(inputs, outputs, examples=None, **kwargs):
                 initialize a registered_gradio_functons
                 if not already initialize.
                 """
-                assert len(inputs) == func.__code__.co_argcount - 1, "❌ inputs should have the same length as arguments"
+                assert len(inputs) == func.__code__.co_argcount - 1, f"❌ {bcolor.BOLD}{bcolor.FAIL}inputs should have the same length as arguments{bcolor.ENDC}"
 
                 try:
                     self = args[0]
@@ -102,7 +105,7 @@ def register(inputs, outputs, examples=None, **kwargs):
                 """
                 the function is not a class function
                 """
-                assert len(inputs) == func.__code__.co_argcount, "❌ inputs should have the same length as arguments"
+                assert len(inputs) == func.__code__.co_argcount, f"❌ {bcolor.BOLD}{bcolor.FAIL}inputs should have the same length as arguments{bcolor.ENDC}"
 
                 if len(args) == (func.__code__.co_argcount):
                     return func(*args, **wargs)

backend/test/test.py

@@ -0,0 +1,39 @@
+import unittest
+import sys
+sys.path.insert(0, "../src/resources")
+from module import GradioModule, register
+
+@GradioModule
+class test:
+        @register(inputs=["text"], outputs=["text"])
+        def Hello(self, name):
+            return f"Hello, {name}."
+
+        @register(inputs=["text"], outputs=["text"])
+        def Goodbye(self, name):
+            return f"Goodbye, {name}."
+    
+@register(inputs=["text"], outputs=["text"])
+def Hello(name):
+    return f"Hello, {name}."
+
+@register(inputs=["text"], outputs=["text"])
+def Goodbye(name):
+    return f"Goodbye, {name}."
+
+class GradioFlowTestCase(unittest.TestCase):
+    cls = test()
+    def test_class_func_return(self):
+        self.assertEqual(self.cls.cls.Hello("Luca"), "Hello, Luca.")
+        self.assertEqual(self.cls.cls.Goodbye("Luca"), "Goodbye, Luca.")
+
+
+    def test_func_return(self):
+        self.assertEqual(Hello("Luca"), "Hello, Luca.")
+        self.assertEqual(Goodbye("Luca"), "Goodbye, Luca.")
+  
+    
+
+        
+
+unittest.main()