Update app.py

app.py

@@ -21,6 +21,7 @@ from typing import Optional, Dict, List, Tuple, Any
 
 # --- Local Module Imports ---
 # These modules contain the application's configuration, clients, and utility functions.
+# Note: These files (hf_client.py, etc.) must exist in the same directory.
 from constants import SYSTEM_PROMPTS, AVAILABLE_MODELS, DEMO_LIST
 from hf_client import get_inference_client
 from tavily_search import enhance_query_with_search
@@ -36,10 +37,16 @@ from utils import (
 )
 from deploy import send_to_sandbox, load_project_from_url
 
-# --- Type Aliases for Readability ---
+# --- Type Aliases and Constants ---
 History = List[Tuple[str, str]]
 Model = Dict[str, Any]
-
+DEFAULT_SYSTEM_PROMPT = """
+You are a helpful AI coding assistant. Your primary goal is to generate clean, correct, and efficient code based on the user's request.
+- Follow the user's requirements precisely.
+- If the user asks for a specific language, provide the code in that language.
+- Enclose the final code in a single markdown code block (e.g., ```html ... ```).
+- Do not include any conversational text, apologies, or explanations outside of the code block in your final response.
+"""
 
 # ==============================================================================
 # HELPER FUNCTIONS
@@ -52,7 +59,6 @@ def get_model_details(model_name: str) -> Optional[Model]:
             return model
     return None
 
-
 # ==============================================================================
 # CORE APPLICATION LOGIC
 # ==============================================================================
@@ -69,20 +75,6 @@ def generation_code(
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
     """
     The main function to handle a user's code generation request.
-
-    Args:
-        query: The user's text prompt.
-        file: An uploaded file for context.
-        website_url: A URL to scrape for context.
-        current_model: The dictionary of the currently selected AI model.
-        enable_search: Flag to enable web search for query enhancement.
-        language: The target programming language.
-        history: The existing conversation history.
-        hf_token: The logged-in user's Hugging Face token for billing.
-
-    Returns:
-        A tuple containing the generated code, updated history, preview HTML,
-        and formatted chatbot messages.
     """
     # 1. --- Initialization and Input Sanitization ---
     query = query or ""
@@ -90,9 +82,19 @@ def generation_code(
 
     try:
         # 2. --- System Prompt and Model Selection ---
-        system_prompt = SYSTEM_PROMPTS.get(language, SYSTEM_PROMPTS["default"])
+        system_prompt = SYSTEM_PROMPTS.get(language, DEFAULT_SYSTEM_PROMPT)
         model_id = current_model["id"]
-        provider = current_model["provider"]
+        
+        # Robustly determine the provider based on ID, falling back to a default
+        if model_id.startswith("openai/"):
+            provider = "openai"
+        elif model_id.startswith("gemini/"):
+            provider = "gemini"
+        elif model_id.startswith("fireworks-ai/"):
+            provider = "fireworks-ai"
+        else:
+            # Assume other models are served via standard Hugging Face TGI
+            provider = "huggingface"
 
         # 3. --- Assemble Full Context for the AI ---
         messages = history_to_messages(history, system_prompt)
@@ -115,13 +117,12 @@ def generation_code(
         resp = client.chat.completions.create(
             model=model_id,
             messages=messages,
-            max_tokens=16384,  # Increased token limit for complex code
-            temperature=0.1   # Low temperature for more predictable, stable code
+            max_tokens=16384,
+            temperature=0.1
         )
         content = resp.choices[0].message.content
 
     except Exception as e:
-        # If the API call fails, show a user-friendly error in the chat.
         error_message = f"❌ **An error occurred:**\n\n```\n{str(e)}\n```\n\nPlease check your API keys, model selection, or try again."
         history.append((query, error_message))
         return "", history, "", history_to_chatbot_messages(history)
@@ -133,8 +134,8 @@ def generation_code(
         preview_html = send_to_sandbox(files.get('index.html', ''))
     else:
         clean_code = remove_code_block(content)
-        if history and history[-1][1] not in (None, ""):
-            # Apply search/replace if a previous turn exists
+        # Apply search/replace if a previous turn exists and contains valid code
+        if history and history[-1][1] and not history[-1][1].startswith("❌"):
             code_str = apply_search_replace_changes(history[-1][1], clean_code)
         else:
             code_str = clean_code
@@ -153,10 +154,7 @@ def generation_code(
 
 with gr.Blocks(theme=gr.themes.Soft(), title="AnyCoder - AI Code Generator") as demo:
     # --- State Management ---
-    # Using gr.State to hold non-visible data like conversation history
-    # and the full dictionary of the selected model.
     history_state = gr.State([])
-    # Initialize with the first model from our constants list
     initial_model = AVAILABLE_MODELS[0]
     model_state = gr.State(initial_model)
 
@@ -181,7 +179,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AnyCoder - AI Code Generator") as
         )
 
         with gr.Accordion("🛠️ Inputs & Context", open=True):
-            prompt_in = gr.Textbox(label="Prompt", lines=3, placeholder="e.g., 'Create a dark-themed login form with a spinning loader.'")
+            prompt_in = gr.Textbox(label="Prompt", lines=3, placeholder="e.g., 'Create a dark-themed login form.'")
             file_in = gr.File(label="📎 Attach File (Optional)", type="filepath")
             url_site = gr.Textbox(label="🌐 Scrape Website (Optional)", placeholder="https://example.com")
         
@@ -211,19 +209,16 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AnyCoder - AI Code Generator") as
 # EVENT WIRING
 # ==============================================================================
 
-# Update the model_state when the user selects a new model from the dropdown.
 def on_model_change(model_name: str) -> Dict:
+    """Updates the model_state when the user selects a new model."""
     model_details = get_model_details(model_name)
     return model_details or initial_model
 model_dd.change(fn=on_model_change, inputs=[model_dd], outputs=[model_state])
 
-# Update the syntax highlighting when the language changes.
 language_dd.change(fn=lambda lang: gr.Code(language=lang), inputs=[language_dd], outputs=[code_out])
 
-# The main event listener for the "Generate" button.
 gen_btn.click(
     fn=generation_code,
-    # Note: `hf_token` is passed automatically by Gradio and is not listed here.
     inputs=[
         prompt_in, file_in, url_site,
         model_state, search_chk, language_dd, history_state
@@ -231,12 +226,13 @@ gen_btn.click(
     outputs=[code_out, history_state, preview_out, chat_out]
 )
 
-# Clear button functionality to reset the interface.
 def clear_session():
+    """Resets the UI components and state to their initial values."""
     return "", [], "", [], None, ""
 clr_btn.click(
     fn=clear_session,
-    outputs=[prompt_in, history_state, preview_out, chat_out, file_in, url_site]
+    outputs=[prompt_in, history_state, preview_out, chat_out, file_in, url_site],
+    queue=False
 )
 
 
@@ -245,5 +241,4 @@ clr_btn.click(
 # ==============================================================================
 
 if __name__ == '__main__':
-    # Launch the Gradio app with queuing enabled for handling multiple users.
     demo.queue().launch()