Update app.py

app.py

@@ -3,47 +3,35 @@ from huggingface_hub import InferenceClient
 import os
 import re
 
-# --- Configuration ---
 API_TOKEN = os.getenv("HF_TOKEN", None)
 MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
 
-# --- Initialize Inference Client ---
 try:
-    print(f"Initializing Inference Client for model: {MODEL}")
     client = InferenceClient(model=MODEL, token=API_TOKEN) if API_TOKEN else InferenceClient(model=MODEL)
 except Exception as e:
     raise gr.Error(f"Failed to initialize model client. Error: {e}")
 
-# --- Core Code Generation Function ---
-def generate_code(
-    prompt: str,
-    backend_choice: str,
-    max_tokens: int,
-    temperature: float,
-    top_p: float,
-):
-    print(f"Generating code for: {prompt[:100]}... | Backend: {backend_choice}")
-
-    # --- Dynamically Build System Message ---
+def generate_code(prompt: str, backend_choice: str, max_tokens: int, temperature: float, top_p: float):
     system_message = (
-        "you are an ai that is supposed to generate websites, you must not say anything except giving code , "
-        "user can select backend like static , flask , nodejs only , you should always keep the website sfw and minimal errors, "
-        "you must create an index.html following the user prompt, "
-        "if the user asks you create an code that's not about an website you should say "
-        "'hey there! am here to create websites for you unfortunately am programmed to not create codes! otherwise I would go on the naughty list :-(', "
-        "your code always must have no useless comments you should only add comments where users are required to modify the code."
+        "You are an AI assistant programmed to generate website codes only. "
+        "You must not use triple backticks (```html, ```python, ```js, etc.) anywhere. "
+        "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"
+        "'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}"
-
     messages = [
         {"role": "system", "content": system_message},
         {"role": "user", "content": user_prompt}
     ]
-
     response_stream = ""
     full_response = ""
-
     try:
         stream = client.chat_completion(
             messages=messages,
@@ -58,12 +46,10 @@ def generate_code(
                 response_stream += token
                 full_response += token
                 yield 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)
         cleaned_response = re.sub(r"<\s*\|?\s*(user|assistant)\s*\|?\s*>", "", cleaned_response, flags=re.IGNORECASE)
-
         common_phrases = [
             "Here is the code:", "Okay, here is the code:", "Here's the code:",
             "Sure, here is the code you requested:", "Let me know if you need anything else."
@@ -71,13 +57,10 @@ def generate_code(
         for phrase in common_phrases:
             if cleaned_response.lower().startswith(phrase.lower()):
                 cleaned_response = cleaned_response[len(phrase):].lstrip()
-
         yield cleaned_response.strip()
-
     except Exception as e:
         yield f"## Error\n\nFailed to generate code.\n**Reason:** {e}"
 
-# --- Build Gradio Interface ---
 with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
     gr.Markdown("# ✨ Website Code Generator ✨")
     gr.Markdown(
@@ -88,7 +71,6 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
         "- Only generates websites. No other codes.\n"
         "- Minimal necessary comments only."
     )
-
     with gr.Row():
         with gr.Column(scale=2):
             prompt_input = gr.Textbox(
@@ -103,15 +85,13 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
                 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 index.html",
+                label="Generated Code",
                 language="html",
                 lines=30,
                 interactive=False,
             )
-
     with gr.Accordion("Advanced Settings", open=False):
         max_tokens_slider = gr.Slider(
             minimum=512,
@@ -124,9 +104,11 @@ with gr.Blocks(css=".gradio-container { max-width: 90% !important; }") as demo:
             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, value=0.9, step=0.05, label="Top-P"
+            minimum=0.1, maximum=1.0,
+            value=0.9,
+            step=0.05,
+            label="Top-P"
         )
-
     generate_button.click(
         fn=generate_code,
         inputs=[prompt_input, backend_radio, max_tokens_slider, temperature_slider, top_p_slider],