Update app.py

app.py

@@ -5,6 +5,7 @@ import re
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 def parse_files(raw_response):
+    # Parse files and code from raw AI output
     pattern = re.compile(r"(?:(?:\n|^)([\w\-.\/\\]+)\n)(.+?)(?=\n[\w\-.\/\\]+\n|\Z)", re.DOTALL)
     return pattern.findall(raw_response)
 
@@ -53,11 +54,11 @@ def on_generate(prompt, backend, system_message, max_tokens, temperature, top_p)
         tabs.append(gr.TabItem(label=name.strip(), elem_id=name.strip(), children=[
             gr.Code(value=content.strip(), language=lang)
         ]))
-    return gr.Group(tabs)
+    return tabs  # Return tabs directly
 
 with gr.Blocks() as demo:
     gr.Markdown("### Website Generator (Static / Flask / Node.js)")
-    
+
     with gr.Row():
         prompt = gr.Textbox(label="Describe your website", placeholder="E.g. a portfolio site with dark mode")
         backend = gr.Dropdown(["Static", "Flask", "Node.js"], value="Static", label="Backend")
@@ -68,17 +69,17 @@ with gr.Blocks() as demo:
         temperature = gr.Slider(0.1, 2.0, value=0.7, label="Temperature")
         top_p = gr.Slider(0.1, 1.0, value=0.95, label="Top-p")
 
-    output = gr.Group()
+    out_tabs = gr.Tabs()  # Define the Tabs component
+
     generate_button = gr.Button("Generate Code")
 
+    # Button click action
     generate_button.click(
         on_generate,
         inputs=[prompt, backend, system_message, max_tokens, temperature, top_p],
-        outputs=output,
+        outputs=out_tabs,  # Output goes directly to the Tabs component
         show_progress=True
     )
 
-    output.render()
-
 if __name__ == "__main__":
     demo.launch()