new file: actionFunctions.py

new file: app.py
new file: components.py
new file: models/__init__.py
new file: models/base.py
new file: models/realistic_vision_v6b1.py
new file: models/sdxl.py
new file: requirements.txt

actionFunctions.py

@@ -0,0 +1,26 @@
+from models.base import Model
+import gradio as gr
+
+model = None
+predict = None
+
+def load_model(model_name):
+    
+    global model
+    model = Model(model_name)
+    global predict
+    predict = model.predict
+
+    return model.getModelState()
+
+def fn_gen_text_to_text(x):
+    text = predict
+    return text
+
+def fn_gen_text_to_image(x):
+    image = predict(x)
+    return image
+
+def fn_gen_image_to_text(x):
+    text = predict(x)
+    return text

app.py

@@ -0,0 +1,30 @@
+import numpy as np
+import gradio as gr
+from components import *
+from actionFunctions import *
+
+
+def create_interface():
+    
+    with gr.Blocks(
+        theme=gr.themes.Monochrome(),
+        css="footer{display:none !important};"
+    ) as demo:
+        
+        gr.Markdown("Welcome to tangibleAI")
+
+
+        #TEXT-TO-TEXT GENERATOR INTERFACE
+        ttt_interface()
+
+        #TEXT-TO-IMAGE GENERATOR INTERFACE
+        tti_interface()
+
+        #IMAGE-TO-TEXT GENERATOR INTERFACE
+        itt_interface()
+        
+        return demo
+
+if __name__ == '__main__':
+    demo = create_interface()
+    demo.launch()

components.py

@@ -0,0 +1,50 @@
+import gradio as gr
+from actionFunctions import *
+
+def ttt_interface():
+    
+    with gr.Tab("Text-to-Text"):
+
+        with gr.Row():
+            ttt_text_input = gr.TextArea(label="Input Text")
+            ttt_text_output = gr.TextArea(label="Generated Text")
+
+        with gr.Column():
+            ttt_models = gr.Dropdown(["Model 1","Model 2", "Model 3"], label='Select a Model')
+            ttt_models.change(load_model, ttt_models, None)
+
+        with gr.Row():
+            text_to_text_clear_button = gr.Button("Clear")
+            text_to_text_gen_button = gr.Button("Generate")
+
+            text_to_text_clear_button.click(fn_gen_text_to_text, None, ttt_text_output)
+            text_to_text_gen_button.click(fn_gen_text_to_text, ttt_text_input, ttt_text_output)
+
+def tti_interface():
+    with gr.Tab("Text-to-Image"):
+
+        with gr.Column():
+            tti_image_output = gr.Image(label="Generated Image")
+            tti_text_input = gr.Textbox(label="Input Text")
+
+        with gr.Column():
+            tti_models = gr.Dropdown(["SDXL","REALV6B1"], label='Select a Model')
+            output  = gr.Textbox(label='Progress Output')
+            tti_models.change(load_model, tti_models, output)
+
+            text_to_image_gen_button = gr.Button("Generate")
+            text_to_image_gen_button.click(fn_gen_text_to_image, tti_text_input, tti_image_output)
+
+def itt_interface():
+    with gr.Tab("Image-to-text"):
+
+        with gr.Row():
+            itt_image_input = gr.Image(label="Input Image")
+            itt_text_output = gr.Textbox(label="Generated Text")
+
+        with gr.Column():
+            itt_models = gr.Dropdown(["Model 1","Model 2", "Model 3"], label='Select a Model')
+            itt_models.change(load_model, itt_models, None)
+
+            image_to_text_gen_button = gr.Button("Generate")
+            image_to_text_gen_button.click(fn_gen_image_to_text, itt_image_input, itt_text_output)

models/base.py

@@ -0,0 +1,29 @@
+from .realistic_vision_v6b1 import RealV6B1
+from .sdxl import SDXL
+
+class Model:
+
+    def __init__(self, modelName):
+        self.modelName = modelName
+
+        if self.modelName == 'SDXL':
+             self.modelObj = SDXL()
+             self.model = self.modelObj.load_model()
+
+        elif self.modelName == 'REALV6B1':
+            self.modelObj = RealV6B1()
+            self.model = self.modelObj.load_model()
+
+        else:
+            self.modelObj = None
+            self.model = None
+
+    def getModelState(self):
+        if self.modelObj is None:
+            return "Model Not Loaded"
+        
+        else:
+            return "Model Loaded"
+    
+    def predict(self, prompt):
+        return self.modelObj.predict(self.model, prompt)

models/realistic_vision_v6b1.py

@@ -0,0 +1,17 @@
+from diffusers import DiffusionPipeline
+
+class RealV6B1:
+    
+    def __init__(self):
+        pass
+
+    def load_model(self):
+        pipeline = DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE",)
+        pipeline.safety_checker = None
+
+        return pipeline
+
+    def predict(self, pipeline, prompt):
+        images = pipeline(prompt=prompt,).images[0]
+
+        return images

models/sdxl.py

@@ -0,0 +1,18 @@
+from diffusers import DiffusionPipeline
+
+
+class SDXL:
+
+    def __init__(self):
+        pass
+
+    def load_model(self):
+        pipeline = DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE",)
+        pipeline.safety_checker = None
+
+        return pipeline
+
+    def predict(self, pipeline, prompt):
+        images = pipeline(prompt=prompt,).images[0]
+        
+        return images

requirements.txt

@@ -0,0 +1,6 @@
+gradio
+torch
+transformers
+safetensors
+diffusers
+accelerate