Update app.py

app.py

@@ -14,15 +14,11 @@ except Exception as e:
 def generate_code(prompt: str, backend_choice: str, max_tokens: int, temperature: float, top_p: float):
     system_message = (
         "You are an AI assistant programmed to generate website codes only. "
-        "You must not use triple backticks (```html, ```python, ```js, etc.) anywhere. "
+        "You must not use triple backticks (```html, ```python, etc.). "
         "If multiple files are needed, separate them clearly using:\n"
-        "{index.html}\n.TAB{NAME=nextfile.ext}\n{anotherfile.ext}\n"
-        "You must not add unnecessary comments; only critical modification instructions are allowed. "
-        "Only generate code and nothing else — no explanations, no phrases like 'Here is the code'. "
-        "The user will select backend type (Static / Flask / Node.js), and you must generate accordingly: "
-        "- For Static: simple index.html.\n"
-        "- For Flask or Node.js: include minimal backend scripts with index.html frontend.\n"
-        "If the user requests non-website code, reply with:\n"
+        "TAB.NAME={filename}\n"
+        "Only generate code. No explanations, no phrases like 'Here is the code'. "
+        "If user asks non-website code, reply:\n"
         "'hey there! am here to create websites for you unfortunately am programmed to not create codes! otherwise I would go on the naughty list :-('."
     )
     user_prompt = f"USER_PROMPT = {prompt}\nUSER_BACKEND = {backend_choice}"
@@ -45,7 +41,7 @@ def generate_code(prompt: str, backend_choice: str, max_tokens: int, temperature
             if isinstance(token, str):
                 response_stream += token
                 full_response += token
-                yield response_stream
+                yield {"raw": response_stream}
         cleaned_response = full_response.strip()
         cleaned_response = re.sub(r"^\s*```[a-z]*\s*\n?", "", cleaned_response)
         cleaned_response = re.sub(r"\n?\s*```\s*$", "", cleaned_response)
@@ -57,19 +53,36 @@ def generate_code(prompt: str, backend_choice: str, max_tokens: int, temperature
         for phrase in common_phrases:
             if cleaned_response.lower().startswith(phrase.lower()):
                 cleaned_response = cleaned_response[len(phrase):].lstrip()
-        yield cleaned_response.strip()
+        yield {"final": cleaned_response.strip()}
     except Exception as e:
-        yield f"## Error\n\nFailed to generate code.\n**Reason:** {e}"
+        yield {"error": f"## Error\n\nFailed to generate code.\n**Reason:** {e}"}
+
+def display_output(output_dict):
+    if "error" in output_dict:
+        return gr.Markdown(output_dict["error"])
+    if "raw" in output_dict:
+        return gr.Markdown("Generating...")
+    cleaned_text = output_dict["final"]
+    file_splits = re.split(r'TAB\.NAME=\{(.+?)\}', cleaned_text)
+    components = []
+    if not file_splits[0].strip():
+        file_splits = file_splits[1:]
+    for i in range(0, len(file_splits), 2):
+        filename = file_splits[i].strip()
+        content = file_splits[i+1].strip() if (i+1) < len(file_splits) else ""
+        components.append(gr.Markdown(f"### {filename}"))
+        components.append(gr.Code(value=content, label=filename, language="html" if filename.endswith(".html") else "python" if filename.endswith(".py") else "javascript" if filename.endswith(".js") else "text", interactive=False))
+    return components
 
 with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
     gr.Markdown("# ✨ Website Code Generator ✨")
     gr.Markdown(
-        "Describe the website you want. The AI will generate a **single-file** `index.html` website.\n\n"
+        "Describe the website you want. The AI will generate website code.\n\n"
         "**Rules:**\n"
         "- Backend hint (Static / Flask / Node.js).\n"
         "- Always fully SFW and minimal errors.\n"
-        "- Only generates websites. No other codes.\n"
-        "- Minimal necessary comments only."
+        "- Only websites. No other codes.\n"
+        "- Multiple files use TAB.NAME={filename}."
     )
     with gr.Row():
         with gr.Column(scale=2):
@@ -82,16 +95,10 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
                 ["Static", "Flask", "Node.js"],
                 label="Backend Context",
                 value="Static",
-                info="Hint only. Always generates only index.html."
             )
             generate_button = gr.Button("✨ Generate Website Code", variant="primary")
         with gr.Column(scale=3):
-            code_output = gr.Code(
-                label="Generated Code",
-                language="html",
-                lines=30,
-                interactive=False,
-            )
+            output_display = gr.Group()
     with gr.Accordion("Advanced Settings", open=False):
         max_tokens_slider = gr.Slider(
             minimum=512,
@@ -101,10 +108,14 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
             label="Max New Tokens"
         )
         temperature_slider = gr.Slider(
-            minimum=0.1, maximum=1.2, value=0.7, step=0.1, label="Temperature"
+            minimum=0.1, maximum=1.2,
+            value=0.7,
+            step=0.1,
+            label="Temperature"
         )
         top_p_slider = gr.Slider(
-            minimum=0.1, maximum=1.0,
+            minimum=0.1,
+            maximum=1.0,
             value=0.9,
             step=0.05,
             label="Top-P"
@@ -112,7 +123,11 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
     generate_button.click(
         fn=generate_code,
         inputs=[prompt_input, backend_radio, max_tokens_slider, temperature_slider, top_p_slider],
-        outputs=code_output,
+        outputs=output_display,
+    ).then(
+        fn=display_output,
+        inputs=output_display,
+        outputs=output_display,
     )
 
 if __name__ == "__main__":