Update app.py

app.py

@@ -15,7 +15,6 @@ from typing import Optional, Dict, List, Tuple, Any
 from constants import (
     HTML_SYSTEM_PROMPT,
     TRANSFORMERS_JS_SYSTEM_PROMPT,
-    GENERIC_SYSTEM_PROMPT,
     AVAILABLE_MODELS,
     DEMO_LIST,
 )
@@ -37,7 +36,7 @@ from deploy import send_to_sandbox
 History = List[Tuple[str, str]]
 Model = Dict[str, Any]
 
-# --- Supported languages dropdown & syntax highlighting ---
+# --- Supported languages for dropdown ---
 SUPPORTED_LANGUAGES = [
     "python", "c", "cpp", "markdown", "latex", "json", "html", "css",
     "javascript", "jinja2", "typescript", "yaml", "dockerfile", "shell",
@@ -64,17 +63,21 @@ def generation_code(
     query = query or ""
     history = history or []
     try:
-        # choose system prompt
+        # Choose system prompt based on language
         if language == "html":
             system_prompt = HTML_SYSTEM_PROMPT
         elif language == "transformers.js":
             system_prompt = TRANSFORMERS_JS_SYSTEM_PROMPT
         else:
-            system_prompt = GENERIC_SYSTEM_PROMPT.format(language=language)
+            # Generic fallback prompt
+            system_prompt = (
+                f"You are an expert {language} developer. "
+                f"Write clean, idiomatic {language} code based on the user's request."
+            )
 
         model_id = current_model["id"]
-        # pick provider
-        if model_id.startswith("openai/") or model_id in {"gpt-4","gpt-3.5-turbo"}:
+        # Determine provider
+        if model_id.startswith("openai/") or model_id in {"gpt-4", "gpt-3.5-turbo"}:
             provider = "openai"
         elif model_id.startswith("gemini/") or model_id.startswith("google/"):
             provider = "gemini"
@@ -83,20 +86,20 @@ def generation_code(
         else:
             provider = "auto"
 
-        # assemble messages
+        # Build message history
         msgs = history_to_messages(history, system_prompt)
-        ctx = query
+        context = query
         if file:
             ftext = extract_text_from_file(file)
-            ctx += f"\n\n[Attached file]\n{ftext[:5000]}"
+            context += f"\n\n[Attached file]\n{ftext[:5000]}"
         if website_url:
             wtext = extract_website_content(website_url)
             if not wtext.startswith("Error"):
-                ctx += f"\n\n[Website content]\n{wtext[:8000]}"
-        final_q = enhance_query_with_search(ctx, enable_search)
+                context += f"\n\n[Website content]\n{wtext[:8000]}"
+        final_q = enhance_query_with_search(context, enable_search)
         msgs.append({"role": "user", "content": final_q})
 
-        # call model
+        # Call the model
         client = get_inference_client(model_id, provider)
         resp = client.chat.completions.create(
             model=model_id,
@@ -111,7 +114,7 @@ def generation_code(
         history.append((query, err))
         return "", history, "", history_to_chatbot_messages(history)
 
-    # post-process
+    # Process model output
     if language == "transformers.js":
         files = parse_transformers_js_output(content)
         code = format_transformers_js_output(files)
@@ -174,7 +177,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), css=CUSTOM_CSS, title="
         with gr.Column(scale=2):
             with gr.Tabs():
                 with gr.Tab("💻 Code"):
-                    code_out = gr.Code(language=None, interactive=True)
+                    code_out = gr.Code(language="html", interactive=True)
                 with gr.Tab("👁️ Live Preview"):
                     preview_out = gr.HTML()
                 with gr.Tab("📜 History"):