Update app.py

app.py

@@ -16,7 +16,7 @@ from utils import (
     parse_transformers_js_output,
     format_transformers_js_output
 )
-from deploy import send_to_sandbox, handle_load_project
+from deploy import send_to_sandbox, load_project_from_url
 
 # Type aliases
 History = List[Tuple[str, str]]
@@ -29,107 +29,134 @@ def generation_code(
     website_url: Optional[str],
     _setting: Dict[str, str],
     _history: Optional[History],
-    _current_model: Dict,
+    _current_model_name: str,
     enable_search: bool,
     language: str,
     provider: str
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
-
+    # Initialize
     if query is None:
         query = ''
     if _history is None:
         _history = []
 
+    # System prompt and history
     system_prompt = _setting.get('system', HTML_SYSTEM_PROMPT)
     messages = history_to_messages(_history, system_prompt)
 
+    # File input
     if file:
-        file_text = extract_text_from_file(file)
-        if file_text:
-            query += f"\n\n[Reference file content below]\n{file_text[:5000]}"
+        text = extract_text_from_file(file)
+        query += f"\n\n[File content]\n{text[:5000]}"
 
+    # Website input
     if website_url:
-        website_text = extract_website_content(website_url)
-        if not website_text.startswith("Error"):
-            query += f"\n\n[Website content below]\n{website_text[:8000]}"
+        text = extract_website_content(website_url)
+        if not text.startswith('Error'):
+            query += f"\n\n[Website]\n{text[:8000]}"
 
+    # Web search enhancement
     final_query = enhance_query_with_search(query, enable_search)
     messages.append({'role': 'user', 'content': final_query})
 
-    client = get_inference_client(_current_model['id'], provider)
-    completion = client.chat.completions.create(
-        model=_current_model['id'],
+    # Model inference
+    client = get_inference_client(_current_model_name, provider)
+    resp = client.chat.completions.create(
+        model=_current_model_name,
         messages=messages,
         max_tokens=10000
     )
-    content = completion.choices[0].message.content
+    content = resp.choices[0].message.content
 
-    has_existing = bool(_history and _history[-1][1])
+    # Post-processing
+    has_existing = bool(_history)
     if language == 'transformers.js':
         files = parse_transformers_js_output(content)
         code_str = format_transformers_js_output(files)
-        sandbox_html = send_to_sandbox(files['index.html'])
+        preview_html = send_to_sandbox(files['index.html'])
     else:
         clean = remove_code_block(content)
-        if has_existing and not clean.strip().startswith('<!DOCTYPE'):
+        if has_existing:
             clean = apply_search_replace_changes(_history[-1][1], clean)
         code_str = clean
-        sandbox_html = send_to_sandbox(clean) if language == 'html' else ''
+        preview_html = send_to_sandbox(clean) if language == 'html' else ''
 
+    # Update history
     new_history = _history + [(query, code_str)]
     chat_msgs = history_to_chatbot_messages(new_history)
 
-    return code_str, new_history, sandbox_html, chat_msgs
+    return code_str, new_history, preview_html, chat_msgs
 
-with gr.Blocks(
-    theme=gr.themes.Base(),
-    title="AnyCoder - AI Code Generator"
-) as demo:
+# Build UI
+with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as demo:
+    # State
     history_state = gr.State([])
     setting_state = gr.State({'system': HTML_SYSTEM_PROMPT})
-    current_model = gr.State(AVAILABLE_MODELS[9])
+    model_state = gr.State(AVAILABLE_MODELS[0]['id'])
 
     with gr.Sidebar():
-        gr.LoginButton()
-        load_project_url = gr.Textbox(label="Hugging Face Space URL")
-        load_project_btn = gr.Button("Import Project")
-        load_project_status = gr.Markdown(visible=False)
-
-        input_box = gr.Textbox(label="What to build?", lines=3)
-        language_dropdown = gr.Dropdown(choices=["html", "python", "transformers.js"], value="html")
-        website_input = gr.Textbox(label="Website URL")
-        file_input = gr.File(label="Reference file")
-        image_input = gr.Image(label="Design image")
-        search_toggle = gr.Checkbox(label="Web search")
-        model_dropdown = gr.Dropdown(choices=[m['name'] for m in AVAILABLE_MODELS], value=AVAILABLE_MODELS[9]['name'])
-
-        generate_btn = gr.Button("Generate")
-        clear_btn = gr.Button("Clear")
+        gr.Markdown("## AnyCoder AI")
+        # Load project
+        url_in = gr.Textbox(label="Load HF Space URL", placeholder="https://huggingface.co/spaces/user/project")
+        load_btn = gr.Button("Import Project")
+        load_status = gr.Markdown(visible=False)
+
+        gr.Markdown("---")
+        # Inputs
+        prompt_in = gr.Textbox(label="Prompt", lines=3)
+        file_in = gr.File(label="Reference file")
+        image_in = gr.Image(label="Design image")
+        url_site = gr.Textbox(label="Website URL")
+        search_chk = gr.Checkbox(label="Enable Web Search")
+        language_dd = gr.Dropdown(choices=["html","python","transformers.js"], value="html", label="Language")
+        model_dd = gr.Dropdown(
+            choices=[m['name'] for m in AVAILABLE_MODELS],
+            value=AVAILABLE_MODELS[0]['name'],
+            label="Model"
+        )
+        gen_btn = gr.Button("Generate")
+        clr_btn = gr.Button("Clear")
 
     with gr.Column():
         with gr.Tabs():
             with gr.Tab("Code"):
-                code_output = gr.Code(label="Generated code")
+                code_out = gr.Code(label="Generated Code")
             with gr.Tab("Preview"):
-                preview = gr.HTML(label="Live preview")
+                preview_out = gr.HTML(label="Live Preview")
             with gr.Tab("History"):
-                history_output = gr.Chatbot()
+                chat_out = gr.Chatbot(label="History")
+
+    # Events
+    load_btn.click(
+        fn=lambda u: load_project_from_url(u),
+        inputs=[url_in],
+        outputs=[load_status, code_out, preview_out, url_in, history_state, chat_out]
+    )
 
-    load_project_btn.click(
-        fn=handle_load_project,
-        inputs=[load_project_url],
-        outputs=[load_project_status, code_output, preview, load_project_url, history_state, history_output]
+    def on_model_change(name):
+        # find id by name
+        for m in AVAILABLE_MODELS:
+            if m['name'] == name:
+                return m['id']
+        return AVAILABLE_MODELS[0]['id']
+
+    model_dd.change(
+        fn=on_model_change,
+        inputs=[model_dd],
+        outputs=[model_state]
     )
 
-    generate_btn.click(
+    gen_btn.click(
         fn=generation_code,
-        inputs=[input_box, image_input, file_input, website_input,
-                setting_state, history_state, current_model,
-                search_toggle, language_dropdown, gr.State('auto')],
-        outputs=[code_output, history_state, preview, history_output]
+        inputs=[prompt_in, image_in, file_in, url_site,
+                setting_state, history_state, model_state,
+                search_chk, language_dd, gr.State('auto')],
+        outputs=[code_out, history_state, preview_out, chat_out]
     )
 
-    clear_btn.click(lambda: ([], [], "", []), outputs=[history_state, history_output, preview, code_output])
+    clr_btn.click(lambda: ([], [], "", []),
+                  outputs=[history_state, chat_out, preview_out, code_out])
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     demo.queue().launch()
+```