Update app.py

app.py

@@ -27,11 +27,15 @@ def parse_files(raw_response):
     if not cleaned_files and raw_response.strip():
         if any(c in raw_response for c in ['<','>','{','}',';','(',')']):
             lang = "html"
-            if "{" in raw_response and "}" in raw_response and ":" in raw_response: lang = "css"
-            elif "function" in raw_response or "const" in raw_response or "let" in raw_response: lang = "javascript"
+            if "{" in raw_response and "}" in raw_response and ":" in raw_response:
+                lang = "css"
+            elif "function" in raw_response or "const" in raw_response or "let" in raw_response:
+                lang = "javascript"
             default_filename = "index.html"
-            if lang == "css": default_filename = "style.css"
-            elif lang == "javascript": default_filename = "script.js"
+            if lang == "css":
+                default_filename = "style.css"
+            elif lang == "javascript":
+                default_filename = "script.js"
             cleaned_files.append((default_filename, raw_response.strip()))
     return cleaned_files
 
@@ -103,9 +107,6 @@ Start generating the files now.
             gr.TabItem(label="Error", children=[gr.Textbox(value=error_message, label="Generation Error", lines=10)])
         ])
     else:
-        print("\n--- Final Raw AI Response ---")
-        print(full_raw_response)
-        print("--------------------------\n")
         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."
@@ -121,22 +122,31 @@ Start generating the files now.
             if not name or not content:
                 continue
             lang = "plaintext"
-            if name.endswith((".html", ".htm")): lang = "html"
-            elif name.endswith(".css"): lang = "css"
-            elif name.endswith(".js"): lang = "javascript"
-            elif name.endswith(".py"): lang = "python"
-            elif name.endswith(".json"): lang = "json"
-            elif name.endswith(".md"): lang = "markdown"
-            elif name.endswith((".sh", ".bash")): lang = "bash"
-            elif name.endswith((".xml", ".xaml", ".svg")): lang = "xml"
-            elif name.endswith(".yaml") or name.endswith(".yml"): lang = "yaml"
+            if name.endswith((".html", ".htm")):
+                lang = "html"
+            elif name.endswith(".css"):
+                lang = "css"
+            elif name.endswith(".js"):
+                lang = "javascript"
+            elif name.endswith(".py"):
+                lang = "python"
+            elif name.endswith(".json"):
+                lang = "json"
+            elif name.endswith(".md"):
+                lang = "markdown"
+            elif name.endswith((".sh", ".bash")):
+                lang = "bash"
+            elif name.endswith((".xml", ".xaml", ".svg")):
+                lang = "xml"
+            elif name.endswith(".yaml") or name.endswith(".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")])])
+            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)
     yield {
@@ -149,62 +159,21 @@ with gr.Blocks(css=".gradio-container { max-width: 95% !important; }") as demo:
     gr.Markdown("Generates website code based on your description. Raw output streams live, final files appear in tabs below.")
     with gr.Row():
         with gr.Column(scale=2):
-            prompt = gr.Textbox(
-                label="Describe your website",
-                placeholder="E.g., a simple landing page for a coffee shop with sections for menu, about, and contact.",
-                lines=3
-            )
-            backend = gr.Dropdown(
-                ["Static", "Flask", "Node.js"],
-                value="Static",
-                label="Backend Technology"
-            )
+            prompt = gr.Textbox(label="Describe your website", placeholder="E.g., a simple landing page...", lines=3)
+            backend = gr.Dropdown(["Static", "Flask", "Node.js"], value="Static", label="Backend Technology")
             with gr.Accordion("Advanced Options", open=False):
-                system_message = gr.Textbox(
-                    label="Extra instructions for the AI (System Message)",
-                    placeholder="Optional: e.g., 'Use Tailwind CSS for styling', 'Make it responsive'",
-                    value="",
-                    lines=2
-                 )
-                max_tokens = gr.Slider(
-                    minimum=256,
-                    maximum=4096,
-                    value=2048,
-                    step=64,
-                    label="Max Tokens (Output Length)"
-                )
-                temperature = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.5,
-                    value=0.7,
-                    step=0.1,
-                    label="Temperature (Creativity)"
-                )
-                top_p = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.95,
-                    step=0.05,
-                    label="Top-p (Sampling Focus)"
-                )
+                system_message = gr.Textbox(label="Extra instructions for the AI (System Message)", placeholder="Optional", value="", lines=2)
+                max_tokens = gr.Slider(minimum=256, maximum=4096, value=2048, step=64, label="Max Tokens (Output Length)")
+                temperature = gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature (Creativity)")
+                top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (Sampling Focus)")
             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="plaintext",
-                lines=20,
-                interactive=False
-            )
+            live_output = gr.Code(label="Raw AI Stream", language="text", lines=20, interactive=False)
     gr.Markdown("---")
     gr.Markdown("#### Final Generated Files (Tabs)")
     final_tabs = gr.Tabs(elem_id="output_tabs")
-    generate_button.click(
-        stream_and_parse_code,
-        inputs=[prompt, backend, system_message, max_tokens, temperature, top_p],
-        outputs=[live_output, final_tabs],
-        show_progress="hidden"
-    )
+    generate_button.click(stream_and_parse_code, inputs=[prompt, backend, system_message, max_tokens, temperature, top_p], outputs=[live_output, final_tabs], show_progress="hidden")
 
 if __name__ == "__main__":
     demo.launch(debug=True)