placed the same gradio format as osiris

app.py

@@ -120,25 +120,53 @@ def multimodal_and_generation(message, history):
         return streamer
 
     
-def diffusing(prompt,
-              history):
+# def diffusing(prompt,
+#               history):
+#     """
+#     Generates an image using Stable Diffusion based on the input prompt.
+#     """
+    
+#     image_base = base(
+#         prompt=prompt,
+#         num_inference_steps=40,
+#         denoising_end=0.8,
+#         output_type="latent",
+#     ).images
+#     image = refiner(
+#         prompt=prompt,
+#         num_inference_steps=40,
+#         denoising_start=0.8,
+#         image=image_base
+#     ).images[0]
+#     print(f"\n\nImage has been diffused:\n {image}\n\n")
+#     return image
+
+# function to take input and generate text tokena
+@spaces.GPU(duration=120)
+def diffusing(prompt: str,
+              n_steps: int,
+              high_noise_frac: float):
     """
-    Generates an image using Stable Diffusion based on the input prompt.
+    Takes input, passes it into the pipeline, 
+    get the top 5 scores, and ouput those scores into images
     """
-    
+
+    # n_steps = int(n_steps)
+
+    # Generate image based on text
     image_base = base(
         prompt=prompt,
-        num_inference_steps=40,
-        denoising_end=0.8,
-        output_type="latent",
+        num_inference_steps=n_steps,
+        denoising_end=high_noise_frac,
+        output_type="latent"
     ).images
     image = refiner(
         prompt=prompt,
-        num_inference_steps=40,
-        denoising_start=0.8,
+        num_inference_steps=n_steps,
+        denoising_start=high_noise_frac,
         image=image_base
     ).images[0]
-    print(f"\n\nImage has been diffused:\n {image}\n\n")
+
     return image
     
 def check_cuda_availability():
@@ -201,57 +229,92 @@ def bot_comms(message, history):
 chatbot = gr.Chatbot(height=600, label="Chimera AI")
 chat_input = gr.MultimodalTextbox(interactive=True, file_types=["images"], placeholder="Enter your question or upload an image.", show_label=False)
 
-with gr.Blocks(fill_height=True) as demo:
-    gr.Markdown(DESCRIPTION)
+# with gr.Blocks(fill_height=True) as demo:
+#     gr.Markdown(DESCRIPTION)
     
-    # image_output = gr.Image(type="pil", label="Generated Image")
+#     # image_output = gr.Image(type="pil", label="Generated Image")
     
-    # def process_response(message, history):
-    #     response = bot_comms(message, history)
-    #     if isinstance(response, tuple) and len(response) == 2:
-    #         text, image = response
-    #         return text, image
-    #     return response, None
-
-    # chatbot_output = gr.Chatbot(height=600, label="Chimera AI")
-
-    # chat_input.submit(process_response, inputs=[chat_input, chatbot], outputs=[chatbot_output, image_output])
-    if mode_manager.get_mode() == "imagery":
-    #     # Ensure's a unique block ID for image output
-        gr.Interface(
-            fn=diffusing,
-            inputs="text",
-            outputs="image",
-            fill_height=True,
-        )
-    #     with gr.Blocks():
-    #         gr.Interface(
-    #             fn=diffusing,
-    #             inputs='text',
-    #             outputs='image',
-    #             fill_height=True,
-    #         )
-    #         # Customize chatinterface to handle tuples
-    #         # def custom_fn(*args, **kwargs):
-    #         #     result = list(bot_comms(*args, **kwargs))
-    #         #     output = []
-    #         #     for item in result:
-    #         #         if isinstance(item, tuple) and isinstance(item[1], Image.Image):
-    #         #             output.append((item[0], None))
-    #         #             output.append((None, item[1]))
-    #         #         else:
-    #         #             output.append(item)
-    #         #     return output
-    # else:
-    #     # Unique block ID for chat interface
-    #     with gr.Blocks():
-    #         gr.ChatInterface(
-    #             fn=bot_comms,
-    #             chatbot=chatbot,
-    #             fill_height=True,
-    #             multimodal=True,
-    #             textbox=chat_input,
-    #         )
+#     # def process_response(message, history):
+#     #     response = bot_comms(message, history)
+#     #     if isinstance(response, tuple) and len(response) == 2:
+#     #         text, image = response
+#     #         return text, image
+#     #     return response, None
+
+#     # chatbot_output = gr.Chatbot(height=600, label="Chimera AI")
+
+#     # chat_input.submit(process_response, inputs=[chat_input, chatbot], outputs=[chatbot_output, image_output])
+#     if mode_manager.get_mode() == "imagery":
+#     #     # Ensure's a unique block ID for image output
+#         gr.Interface(
+#             fn=diffusing,
+#             inputs="text",
+#             outputs="image",
+#             fill_height=True,
+#         )
+#     #     with gr.Blocks():
+#     #         gr.Interface(
+#     #             fn=diffusing,
+#     #             inputs='text',
+#     #             outputs='image',
+#     #             fill_height=True,
+#     #         )
+#     #         # Customize chatinterface to handle tuples
+#     #         # def custom_fn(*args, **kwargs):
+#     #         #     result = list(bot_comms(*args, **kwargs))
+#     #         #     output = []
+#     #         #     for item in result:
+#     #         #         if isinstance(item, tuple) and isinstance(item[1], Image.Image):
+#     #         #             output.append((item[0], None))
+#     #         #             output.append((None, item[1]))
+#     #         #         else:
+#     #         #             output.append(item)
+#     #         #     return output
+#     # else:
+#     #     # Unique block ID for chat interface
+#     #     with gr.Blocks():
+#     #         gr.ChatInterface(
+#     #             fn=bot_comms,
+#     #             chatbot=chatbot,
+#     #             fill_height=True,
+#     #             multimodal=True,
+#     #             textbox=chat_input,
+#     #         )
+
+# if __name__ == "__main__":
+#     demo.launch()
+
+with gr.Blocks(fill_height=True) as demo:
+    gr.Markdown(DESCRIPTION)
+    gr.Interface(
+        fn=osiris,
+        inputs="text",
+        outputs="image",
+        fill_height=True,
+        additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
+        additional_inputs=[
+            gr.Slider(minimum=20,
+                      maximum=100,
+                      step=1,
+                      value=40,
+                      label="Number of Inference Steps",
+                      render=False),
+            gr.Slider(minimum=0.0,
+                      maximum=1.0,
+                      step=0.1,
+                      value=0.8,
+                      label="High Noise Fraction",
+                      render=False),
+        ],
+        examples=[
+            ["A sprawling cyberpunk metropolis at dusk, with towering skyscrapers adorned with neon signs, holographic billboards, and flying cars weaving through the sky. On a crowded street corner, a cybernetically enhanced street artist creates mesmerizing light sculptures with their augmented reality gloves. Rain glistens on the bustling sidewalks as pedestrians with colorful umbrellas rush past."],
+            ["A mystical enchanted forest bathed in twilight, where bioluminescent plants cast an ethereal glow. A crystal-clear waterfall cascades into a shimmering pool, surrounded by ancient trees with twisted roots. A lone elf archer, dressed in elegant green and gold attire, stands on a moss-covered rock, her bow drawn as she watches over the tranquil scene. Ethereal fairies with delicate wings flutter around, leaving trails of sparkling dust in the air."],
+            ["An elaborate steampunk dirigible floating gracefully above a cloud-covered landscape. The airship, with its brass and copper gears, massive steam-powered engines, and ornate Victorian design, cruises past a golden sunset. A distinguished gentleman in a top hat and goggles stands on the observation deck, holding a brass spyglass to his eye as he surveys the horizon. Passengers in vintage attire marvel at the view."],
+            ["A charming, snow-covered log cabin nestled in the heart of a tranquil mountain range during a starry winter night. Smoke gently curls from the stone chimney, and warm light spills from the windows, illuminating the cozy interior. A young woman in a thick woolen coat and a fur-lined hat stands by the front door, holding a lantern that casts a warm glow on the snow. Pine trees, heavy with snow, frame the scene, while the Northern Lights dance across the sky."],
+            ["A vibrant underwater kingdom where a colorful coral reef teems with marine life. Schools of iridescent fish swim through the crystal-clear waters, and a sunken pirate ship, encrusted with barnacles and treasure chests, rests on the sandy seabed. A curious mermaid with flowing turquoise hair and a shimmering silver tail explores the depths, holding a glowing pearl in her hand. Playful dolphins swim around her, and a wise old sea turtle watches from a nearby rock."]
+        ],
+        cache_examples=False
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()