backend refactors

backend/app.py

@@ -2,6 +2,11 @@ from flask import Flask, jsonify, request
 from flask_cors import CORS
 import socket
 import argparse
+import src
+import importlib
+import asyncio 
+from multiprocessing import Process
+import gradio as gr
 
 app = Flask(__name__)
 CORS(app)
@@ -15,13 +20,53 @@ def portConnection(port : int):
     return False
 
 
-global visable, watcher, dog
+global visable
+global process_map
 visable = []
+process_map = {}
 
 @app.route("/")
 def Home():
     return jsonify({"message" :"everything is up amd running... 🚀",})
 
+# def IS_STREAMABLE():
+#     pass
+
+# async def subprocess(metadata):
+#     # fetch the module
+    
+#     try :
+#         DONT_LISTEN_TO = ['GradioModule', 'register']
+#         mode = importlib.import_module(metadata)
+#         fn, names = [], []
+#         for fn_key in dir(mode):
+#             # print(fn_key, type(getattr(mode, fn_key)))
+#             attr = getattr(mode, fn_key)
+#             # p = print(dir(attr)) if fn_key == "Hello_World" else None
+#             if callable(attr) and not fn_key.startswith("__") and  "__decorator__" in dir(attr) and attr.__decorator__ == "__gradio__":
+#                 fn.append((getattr(mode, fn_key)))
+#                 names.append(getattr(mode, fn_key)().__name__)
+#         print(fn, names)
+#     except Exception as e:
+#         print(e) 
+#         raise e 
+    
+    
+#     asyncio.create_task(src.resources.module.tabularGradio(fn, names, listen=2000, metadata=metadata, name=metadata.split('.')[-1]))
+#     return 
+
+# @app.route("/api/append/module" , methods=["POST"])
+# def append_module():
+
+#     current = request.json
+#     # task_cb = Process(target=subprocess, args=(current['module'],))
+#     asyncio.run(subprocess(current['module']))
+#     # run module until 
+#     return jsonify({"executed" : True,
+#                     "process" : "ended" })
+
+
+
 @app.route("/api/append/port" , methods=["POST"])
 def append_port():
     current = request.json
@@ -37,6 +82,15 @@ def remove_port():
                     "ports" : current['port']})
 
 
+# @app.route("/api/remove/module" , methods=["POST"])
+# def remove_module():
+#     current = request.json
+   
+#     visable.remove(current)
+#     process_map[current["kwargs"]["metadata"]].terminate()
+#     return jsonify({"executed" : True,
+#                     "ports" : current['port']})
+
 @app.route("/api/open/ports", methods=["GET"])
 def open_ports():
     return jsonify(visable)
@@ -47,4 +101,6 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("-p", "--port", help="location of flask api port on local host", default=5000)
     args = parser.parse_args()
-    app.run(host="0.0.0.0", port=args.port, debug=True)
+    app.run(host="0.0.0.0", port=args.port, debug=True)
+
+   

backend/src/__init__.py

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

backend/src/demo/demo.py

@@ -2,7 +2,7 @@ import argparse
 import sys
 
 sys.path.insert(0, "../")
-from resources import GradioModule, register, InterLauncher, tabularGradio
+from src.resources import GradioModule, register, InterLauncher, tabularGradio
 from example.examples import Pictionary, FSD, stock_forecast
 
 import gradio as gr

backend/src/demo/module.py

@@ -0,0 +1,11 @@
+import sys
+sys.path.insert(0, "../")
+from src.resources import GradioModule, register
+
+@register(["text"], ["text"], examples=[["Luca Vivona"]])
+def Hello_World(name):
+            return f"👋 Hello {name}, and welcome to Gradio Flow 🤗" 
+
+@register(["number", "number"], ["number"], examples=[[1,1]])
+def add(x, y):
+            return x + y

backend/src/example/__init__.py

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

backend/src/example/examples.py

@@ -7,9 +7,8 @@ import torch
 from torch import nn
 import numpy as np
 import PIL
-
 sys.path.insert(0, "../resources")
-from resources.module import GradioModule, register
+from src.resources.module import GradioModule, register
 
 
 @GradioModule

backend/src/resources/module.py

@@ -337,7 +337,6 @@ def GradioModule(cls):
                 except Exception:
                     print(f"**{bcolor.BOLD}{bcolor.FAIL}CONNECTION ERROR{bcolor.ENDC}** 🐛The api either lost connection or was turned off...🐛")
             return
-
     return Decorator
 
 # console colour changer 

backend/test/app_test.py

@@ -0,0 +1,7 @@
+import requests
+DOCKER_LOCAL_HOST = '0.0.0.0' 
+import gradio as gr
+import asyncio
+if __name__ == "__main__":
+    gr.close_all()
+    requests.post(f"http://{DOCKER_LOCAL_HOST}:{ 2000 }/api/append/module", json={"module" : "src.demo.module"})