Update app.py

app.py

@@ -92,10 +92,9 @@ Start generating the files now.
             if token:
                 full_raw_response += token
                 yield {
-                    live_output: gr.update(value=full_raw_response),
+                    live_output: gr.update(value=full_raw_response)
                 }
     except Exception as e:
-        print(f"Error during AI streaming: {e}")
         error_message = f"Error during AI generation: {e}\n\nPartial Response (if any):\n{full_raw_response}"
         error_occurred = True
         yield {
@@ -104,16 +103,19 @@ Start generating the files now.
         }
     if error_occurred:
         final_tabs_update = gr.Tabs(tabs=[
-            gr.TabItem(label="Error", children=[gr.Textbox(value=error_message, label="Generation Error", lines=10)])
+            gr.TabItem(label="Error", children=[gr.Textbox(value=error_message, lines=10)])
         ])
     else:
         files = parse_files(full_raw_response)
         if not files:
             no_files_msg = "AI finished, but did not return recognizable file content or the response was empty. See raw output above."
             final_tabs_update = gr.Tabs(tabs=[
-                gr.TabItem(label="Output", children=[gr.Textbox(value=no_files_msg, label="Result")])
+                gr.TabItem(label="Output", children=[gr.Textbox(value=no_files_msg)])
             ])
-            yield { live_output: gr.update(value=full_raw_response + "\n\n" + no_files_msg), final_tabs: final_tabs_update }
+            yield {
+                live_output: gr.update(value=full_raw_response + "\n\n" + no_files_msg),
+                final_tabs: final_tabs_update
+            }
             return
         tabs_content = []
         for name, content in files:
@@ -138,17 +140,16 @@ Start generating the files now.
                 lang = "bash"
             elif name.endswith((".xml", ".xaml", ".svg")):
                 lang = "xml"
-            elif name.endswith(".yaml") or name.endswith(".yml"):
+            elif name.endswith((".yaml", ".yml")):
                 lang = "yaml"
             elem_id = f"tab_{re.sub(r'[^a-zA-Z0-9_-]', '_', name)}"
             tab_item = gr.TabItem(label=name, elem_id=elem_id, children=[
                 gr.Code(value=content, language=lang, label=name, interactive=False)
             ])
             tabs_content.append(tab_item)
-        if not tabs_content:
-            final_tabs_update = gr.Tabs(tabs=[gr.TabItem(label="Output", children=[gr.Textbox(value="No valid files generated after filtering.", label="Result")])])
-        else:
-            final_tabs_update = gr.Tabs(tabs=tabs_content)
+        final_tabs_update = gr.Tabs(tabs=tabs_content) if tabs_content else gr.Tabs(tabs=[
+            gr.TabItem(label="Output", children=[gr.Textbox(value="No valid files generated after filtering.")])
+        ])
     yield {
         live_output: gr.update(value=full_raw_response if not error_occurred else error_message),
         final_tabs: final_tabs_update
@@ -169,7 +170,7 @@ with gr.Blocks(css=".gradio-container { max-width: 95% !important; }") as demo:
             generate_button = gr.Button("✨ Generate Code ✨", variant="primary")
         with gr.Column(scale=3):
             gr.Markdown("#### Live Raw Output Stream")
-            live_output = gr.Code(label="Raw AI Stream", language="text", lines=20, interactive=False)
+            live_output = gr.Textbox(label="Raw AI Stream", lines=20, interactive=False)
     gr.Markdown("---")
     gr.Markdown("#### Final Generated Files (Tabs)")
     final_tabs = gr.Tabs(elem_id="output_tabs")