Update app.py

app.py

@@ -1,7 +1,9 @@
 import json
+
 import gradio as gr
 import spaces
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+
 from format.format_output import format_output
 from validate.validate_ingredients import validate_ingredients
 from device.get_device_id import get_device_id
@@ -29,47 +31,31 @@ def generate_recipe(ingredients):
 
     if is_ingredients_valid:
         generated_text = perform_model_inference(ingredients_list)
-        return generated_text
+
+        return {
+            generated_recipe: gr.Markdown(value=generated_text, label="Generated Recipe",
+                                          elem_id="recipe-container", visible=True)
+        }
     else:
-        error_text = "## Invalid ingredients. Please include at least 2 ingredients in a comma-separated list. e.g. brown rice, onions, garlic"
-        return error_text
-
-
-with gr.Blocks(css=""" 
-    #chat-container {
-        display: flex;
-        flex-direction: column;
-        height: 100vh;
-    }
-    .message {
-        padding: 10px;
-        border-radius: 5px;
-    }
-    .user-message {
-        background-color: #e6f7ff;
-        text-align: left;
-    }
-    .bot-message {
-        background-color: #f0f0f0;
-        text-align: left;
-    }
-    .gr-button {
-        background-color: #0073e6 !important;
-        color: white !important;
-        border: none !important;
-        border-radius: 5px !important;
-    }
-""") as recipegen:
-    gr.Markdown("### An AI Model Attempting To Produce Healthier, Diabetic-Friendly Recipes", elem_id="header-sub-heading")
+        error_text = "## Invalid ingredients. Please include at least 2 ingredients in a comma separated list. e.g. brown rice, onions, garlic"
+
+        return {
+            generated_recipe: gr.Markdown(value=error_text, elem_id="recipe-container", visible=True)
+        }
+
+
+with gr.Blocks(css="./css/styles.css") as recipegen:
+    gr.Image("./assets/dut.png", interactive=False, show_share_button=False, show_download_button=False,
+             show_fullscreen_button=False, show_label=False, elem_id="dut-logo", height=256)
+    gr.Markdown("#  Recipe Generator", elem_id="header")
+    gr.Markdown("### An AI Model Attempting To Produce Healthier, Diabetic-Friendly Recipes",
+                elem_id="header-sub-heading")
     gr.Markdown("Start by entering a comma-separated list of ingredients below.", elem_id="header-instructions")
-    
     with gr.Column() as column:
         user_ingredients = gr.Textbox(label="Ingredients", autofocus=True, max_lines=1, elem_id="ingredients-input")
         generate_button = gr.Button(value="Generate")
-        
     with gr.Column():
         generated_recipe = gr.Markdown(visible=True)
-        
     examples = gr.Examples(
         elem_id="examples",
         examples=[
@@ -81,7 +67,6 @@ with gr.Blocks(css="""
         inputs=[user_ingredients]
     )
 
-    # Now we directly generate the recipe when the button is clicked
     generate_button.click(
         fn=generate_recipe,
         inputs=[user_ingredients],