Update app.py

app.py

@@ -1,27 +1,16 @@
-"""
-app.py
+# app.py
 
-Main application file for AnyCoder, a Gradio-based AI code generation tool.
+"""
+Main application file for SHASHA AI, a Gradio-based AI code generation tool.
 
 This application provides a user interface for generating code in various languages
 using different AI models. It supports inputs from text prompts, files, images,
 and websites, and includes features like web search enhancement and live code previews.
-
-Structure:
-- Imports & Configuration: Loads necessary libraries and constants.
-- Helper Functions: Small utility functions supporting the UI logic.
-- Core Application Logic: The main `generation_code` function that handles the AI interaction.
-- UI Layout: Defines the Gradio interface using `gr.Blocks`.
-- Event Wiring: Connects UI components to backend functions.
-- Application Entry Point: Launches the Gradio app.
 """
 
 import gradio as gr
 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
@@ -35,34 +24,33 @@ from utils import (
     parse_transformers_js_output,
     format_transformers_js_output
 )
-from deploy import send_to_sandbox, load_project_from_url
+from deploy import send_to_sandbox
 
-# --- Type Aliases and Constants ---
 History = List[Tuple[str, str]]
 Model = Dict[str, Any]
+
+# Full list of supported languages for syntax highlighting & generation
+SUPPORTED_LANGUAGES = [
+    "python", "c", "cpp", "markdown", "latex", "json", "html", "css",
+    "javascript", "jinja2", "typescript", "yaml", "dockerfile", "shell",
+    "r", "sql", "sql-msSQL", "sql-mySQL", "sql-mariaDB", "sql-sqlite",
+    "sql-cassandra", "sql-plSQL", "sql-hive", "sql-pgSQL", "sql-gql",
+    "sql-gpSQL", "sql-sparkSQL", "sql-esper"
+]
+
 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.
+You are a helpful AI coding assistant. Generate clean, correct, and efficient code based on the user's request.
+- Follow requirements precisely.
+- Enclose final code in a single ```code``` block of the target language.
+- Do not include any explanations outside the code block.
 """
 
-# ==============================================================================
-# HELPER FUNCTIONS
-# ==============================================================================
-
-def get_model_details(model_name: str) -> Optional[Model]:
-    """Finds the full dictionary for a model given its name."""
-    for model in AVAILABLE_MODELS:
-        if model["name"] == model_name:
-            return model
+def get_model_details(name: str) -> Optional[Model]:
+    for m in AVAILABLE_MODELS:
+        if m["name"] == name:
+            return m
     return None
 
-# ==============================================================================
-# CORE APPLICATION LOGIC
-# ==============================================================================
-
 def generation_code(
     query: Optional[str],
     file: Optional[str],
@@ -71,194 +59,129 @@ def generation_code(
     enable_search: bool,
     language: str,
     history: Optional[History],
-    hf_token: str,
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
-    """
-    The main function to handle a user's code generation request.
-    """
-    # 1. --- Initialization and Input Sanitization ---
     query = query or ""
     history = history or []
-
     try:
-        # 2. --- System Prompt and Model Selection ---
         system_prompt = SYSTEM_PROMPTS.get(language, DEFAULT_SYSTEM_PROMPT)
         model_id = current_model["id"]
-        
-        # Robustly determine the provider based on ID, falling back to a default
-        if model_id.startswith("openai/"):
+
+        # pick provider
+        if model_id.startswith("openai/") or model_id in ("gpt-4", "gpt-3.5-turbo"):
             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)
-        context_query = query
-
+        # assemble messages
+        msgs = history_to_messages(history, system_prompt)
+        ctx = query
         if file:
-            text = extract_text_from_file(file)
-            context_query += f"\n\n[Attached File Content]\n{text[:5000]}"
-
+            txt = extract_text_from_file(file)
+            ctx += f"\n\n[File]\n{txt[:5000]}"
         if website_url:
-            text = extract_website_content(website_url)
-            if not text.startswith('Error'):
-                context_query += f"\n\n[Scraped Website Content]\n{text[:8000]}"
-
-        final_query = enhance_query_with_search(context_query, enable_search)
-        messages.append({'role': 'user', 'content': final_query})
+            txt = extract_website_content(website_url)
+            if not txt.startswith("Error"):
+                ctx += f"\n\n[Website]\n{txt[:8000]}"
+        final_q = enhance_query_with_search(ctx, enable_search)
+        msgs.append({"role": "user", "content": final_q})
 
-        # 4. --- AI Model Inference with Robust Error Handling ---
-        client = get_inference_client(model_id, provider, user_token=hf_token)
+        client = get_inference_client(model_id, provider)
         resp = client.chat.completions.create(
             model=model_id,
-            messages=messages,
-            max_tokens=16384,
+            messages=msgs,
+            max_tokens=16000,
             temperature=0.1
         )
         content = resp.choices[0].message.content
 
     except Exception as e:
-        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))
+        err = f"❌ **Error:**\n```\n{e}\n```"
+        history.append((query, err))
         return "", history, "", history_to_chatbot_messages(history)
 
-    # 5. --- Post-process the AI's Output ---
-    if language == 'transformers.js':
+    # post-process
+    if language == "transformers.js":
         files = parse_transformers_js_output(content)
-        code_str = format_transformers_js_output(files)
-        preview_html = send_to_sandbox(files.get('index.html', ''))
+        code = format_transformers_js_output(files)
+        preview = send_to_sandbox(files.get("index.html",""))
     else:
-        clean_code = remove_code_block(content)
-        # Apply search/replace if a previous turn exists and contains valid code
+        clean = remove_code_block(content)
         if history and history[-1][1] and not history[-1][1].startswith("❌"):
-            code_str = apply_search_replace_changes(history[-1][1], clean_code)
+            code = apply_search_replace_changes(history[-1][1], clean)
         else:
-            code_str = clean_code
-        preview_html = send_to_sandbox(code_str) if language == 'html' else ''
-
-    # 6. --- Update History and Final Outputs ---
-    updated_history = history + [(query, code_str)]
-    chat_messages = history_to_chatbot_messages(updated_history)
-
-    return code_str, updated_history, preview_html, chat_messages
-
+            code = clean
+        preview = send_to_sandbox(code) if language == "html" else ""
 
-# ==============================================================================
-# UI LAYOUT & EVENT WIRING
-# ==============================================================================
+    new_hist = history + [(query, code)]
+    chat = history_to_chatbot_messages(new_hist)
+    return code, new_hist, preview, chat
 
-# Custom CSS for a more professional and modern look
+# custom CSS
 CUSTOM_CSS = """
-body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif; }
-#main_title { text-align: center; font-size: 2.5rem; font-weight: 700; color: #1a202c; margin: 1.5rem 0 0.5rem 0; }
-#subtitle { text-align: center; color: #4a5568; margin-bottom: 2.5rem; font-size: 1.1rem; }
-.gradio-container { background-color: #f7fafc; }
-/* Custom styling for the generate button to make it stand out */
-#gen_btn { box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06); }
+body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; }
+#main_title { text-align: center; font-size: 2.5rem; margin: 1.5rem 0 0.5rem; }
+#subtitle { text-align: center; color: #4a5568; margin-bottom: 2.5rem; }
+.gradio-container { background: #f7fafc; }
+#gen_btn { box-shadow: 0 4px 6px rgba(0,0,0,0.1); }
 """
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), title="AnyCoder - AI Code Generator", css=CUSTOM_CSS) as demo:
-    # --- State Management ---
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), css=CUSTOM_CSS, title="Shasha AI") as demo:
     history_state = gr.State([])
-    initial_model = AVAILABLE_MODELS[0]
-    model_state = gr.State(initial_model)
+    initial = AVAILABLE_MODELS[0]
+    model_state = gr.State(initial)
 
-    # --- UI Definition ---
     gr.Markdown("# 🚀 Shasha AI", elem_id="main_title")
     gr.Markdown("Your personal AI partner for generating, modifying, and understanding code.", elem_id="subtitle")
 
-    with gr.Row(equal_height=False):
-        # Left column for controls and inputs
+    with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### 1. Select Model")
-            model_choices = [model["name"] for model in AVAILABLE_MODELS]
-            model_dd = gr.Dropdown(
-                choices=model_choices,
-                value=initial_model["name"],
-                label="AI Model",
-                info="Different models have different strengths."
-            )
+            names = [m["name"] for m in AVAILABLE_MODELS]
+            model_dd = gr.Dropdown(names, value=initial["name"], label="AI Model")
 
             gr.Markdown("### 2. Provide Context")
             with gr.Tabs():
                 with gr.Tab("📝 Prompt"):
-                    prompt_in = gr.Textbox(
-                        label="Your Request",
-                        lines=7,
-                        placeholder="e.g., 'Create a modern, responsive landing page for a SaaS product.'",
-                        show_label=False
-                    )
+                    prompt_in = gr.Textbox(lines=7, placeholder="Describe your request...", show_label=False)
                 with gr.Tab("📄 File"):
-                    file_in = gr.File(label="Attach File (Optional)", type="filepath")
+                    file_in = gr.File(type="filepath")
                 with gr.Tab("🌐 Website"):
-                    url_site = gr.Textbox(label="Scrape Website (Optional)", placeholder="https://example.com")
+                    url_in = gr.Textbox(placeholder="https://example.com")
 
             gr.Markdown("### 3. Configure Output")
-            language_dd = gr.Dropdown(
-                choices=["html", "python", "transformers.js", "sql", "javascript", "css"],
-                value="html",
-                label="Target Language"
-            )
-            search_chk = gr.Checkbox(label="Enable Web Search", info="Enhances AI with real-time info.")
+            lang_dd = gr.Dropdown(SUPPORTED_LANGUAGES, value="html", label="Target Language")
+            search_chk = gr.Checkbox(label="Enable Web Search")
 
             with gr.Row():
-                clr_btn = gr.Button("Clear Session", variant="secondary")
-                gen_btn = gr.Button("Generate Code", variant="primary", elem_id="gen_btn")
+                clr = gr.Button("Clear Session", variant="secondary")
+                gen = gr.Button("Generate Code", variant="primary", elem_id="gen_btn")
 
-        # Right column for outputs
         with gr.Column(scale=2):
-            with gr.Tabs() as main_tabs:
-                with gr.Tab("💻 Code", id="code_tab"):
-                    code_out = gr.Code(label="Generated Code", language="html", interactive=True)
-                with gr.Tab("👁️ Live Preview", id="preview_tab"):
-                    preview_out = gr.HTML(label="Live Preview")
-                with gr.Tab("📜 History", id="history_tab"):
-                    chat_out = gr.Chatbot(label="Conversation History", type="messages")
-
-    # --- Event Wiring ---
-    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
+            with gr.Tabs():
+                with gr.Tab("💻 Code"):
+                    code_out = gr.Code(language=lambda: lang_dd.value, interactive=True)
+                with gr.Tab("👁️ Live Preview"):
+                    preview_out = gr.HTML()
+                with gr.Tab("📜 History"):
+                    chat_out = gr.Chatbot(type="messages")
 
-    model_dd.change(fn=on_model_change, inputs=[model_dd], outputs=[model_state])
-    language_dd.change(fn=lambda lang: gr.update(language=lang), inputs=[language_dd], outputs=[code_out])
+    model_dd.change(lambda n: get_model_details(n) or initial, inputs=[model_dd], outputs=[model_state])
 
-    gen_btn.click(
+    gen.click(
         fn=generation_code,
-        inputs=[
-            prompt_in, file_in, url_site,
-            model_state, search_chk, language_dd, history_state
-        ],
-        outputs=[code_out, history_state, preview_out, chat_out]
+        inputs=[prompt_in, file_in, url_in, model_state, search_chk, lang_dd, history_state],
+        outputs=[code_out, history_state, preview_out, chat_out],
     )
 
-    def clear_session():
-        """Resets all UI components and state to their initial values."""
-        return (
-            "",      # prompt_in
-            None,    # file_in
-            "",      # url_site
-            [],      # history_state
-            "",      # code_out
-            "",      # preview_out
-            []       # chat_out
-        )
-
-    clr_btn.click(
-        fn=clear_session,
-        outputs=[prompt_in, file_in, url_site, history_state, code_out, preview_out, chat_out],
-        queue=False
+    clr.click(
+        fn=lambda: ("", None, "", [], "", "", []),
+        outputs=[prompt_in, file_in, url_in, history_state, code_out, preview_out, chat_out],
+        queue=False,
     )
 
-# ==============================================================================
-# APPLICATION ENTRY POINT
-# ==============================================================================
-
-if __name__ == '__main__':
-    demo.queue().launch()
+if __name__ == "__main__":
+    demo.queue().launch()