Update app.py

app.py

@@ -3,22 +3,17 @@ from huggingface_hub import InferenceClient
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def respond(message, history, system_message, max_tokens, temperature, top_p, backend):
-    forced_system = f"""
+def generate_code(prompt, system_message, max_tokens, temperature, top_p, backend):
+    full_system_prompt = f"""
 You are a code-generation AI. You MUST generate a full website including an index.html file.
-Use only the {backend} backend structure. 
+Use ONLY the {backend} backend structure. 
 Respond ONLY with raw code and file/folder structure. Do NOT explain or add commentary.
-""".strip()
+""".strip() + "\n\n" + system_message
 
-    system_message = forced_system + "\n\n" + system_message
-
-    messages = [{"role": "system", "content": system_message}]
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
-    messages.append({"role": "user", "content": message})
+    messages = [
+        {"role": "system", "content": full_system_prompt},
+        {"role": "user", "content": prompt}
+    ]
 
     response = ""
     for chunk in client.chat_completion(
@@ -26,7 +21,7 @@ Respond ONLY with raw code and file/folder structure. Do NOT explain or add comm
         max_tokens=max_tokens,
         stream=True,
         temperature=temperature,
-        top_p=top_p,
+        top_p=top_p
     ):
         token = chunk.choices[0].delta.content
         if token:
@@ -34,46 +29,25 @@ Respond ONLY with raw code and file/folder structure. Do NOT explain or add comm
             yield response
 
 with gr.Blocks() as demo:
-    gr.Markdown("# WebGen AI\nGenerate a complete website with your selected backend.")
-
-    with gr.Row():
-        system_msg = gr.Textbox(value="You are a helpful assistant.", label="System Message")
-        backend = gr.Dropdown(["Flask", "Static", "Node.js"], value="Static", label="Backend")
+    gr.Markdown("## WebGen AI — One Prompt → Full Website Generator")
 
     with gr.Row():
-        max_tokens = gr.Slider(1, 2048, value=512, label="Max Tokens")
-        temperature = gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature")
-        top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
-
-    chatbot = gr.Chatbot()
-    user_input = gr.Textbox(label="Your Prompt", placeholder="Ask the AI to generate a website...")
-
-    history = []
+        prompt = gr.Textbox(label="Enter Prompt", placeholder="Describe the website you want...")
+        backend = gr.Dropdown(["Flask", "Static", "Node.js"], value="Static", label="Select Backend")
 
-    def chat_submit(message):
-        nonlocal history
-        history.append([message, None])
-        return "", history
+    system_msg = gr.Textbox(value="You are a helpful assistant.", label="System Message")
+    max_tokens = gr.Slider(1, 2048, value=512, label="Max Tokens")
+    temperature = gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature")
+    top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
 
-    send_btn = gr.Button("Send")
+    code_output = gr.Code(label="AI Output (Live)", language="text")
 
-    def run_response(message, system_msg, max_tokens, temperature, top_p, backend):
-        nonlocal history
-        response_generator = respond(message, history, system_msg, max_tokens, temperature, top_p, backend)
-        final_response = ""
-        for chunk in response_generator:
-            final_response = chunk
-            yield history[:-1] + [[message, chunk]]
-        history[-1][1] = final_response
+    run_btn = gr.Button("Generate Code")
 
-    send_btn.click(
-        chat_submit,
-        inputs=[user_input],
-        outputs=[user_input, chatbot]
-    ).then(
-        run_response,
-        inputs=[user_input, system_msg, max_tokens, temperature, top_p, backend],
-        outputs=chatbot
+    run_btn.click(
+        generate_code,
+        inputs=[prompt, system_msg, max_tokens, temperature, top_p, backend],
+        outputs=code_output
     )
 
 if __name__ == "__main__":