Update app.py

app.py

@@ -34,7 +34,7 @@ def generation_code(
     language: str,
     provider: str
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
-    # Initialize
+    # Initialize inputs
     if query is None:
         query = ''
     if _history is None:
@@ -53,7 +53,7 @@ def generation_code(
     if website_url:
         text = extract_website_content(website_url)
         if not text.startswith('Error'):
-            query += f"\n\n[Website]\n{text[:8000]}"
+            query += f"\n\n[Website content]\n{text[:8000]}"
 
     # Web search enhancement
     final_query = enhance_query_with_search(query, enable_search)
@@ -96,19 +96,25 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
 
     with gr.Sidebar():
         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")
+        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'],
@@ -133,8 +139,7 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
         outputs=[load_status, code_out, preview_out, url_in, history_state, chat_out]
     )
 
-    def on_model_change(name):
-        # find id by name
+    def on_model_change(name: str) -> str:
         for m in AVAILABLE_MODELS:
             if m['name'] == name:
                 return m['id']
@@ -148,15 +153,18 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
 
     gen_btn.click(
         fn=generation_code,
-        inputs=[prompt_in, image_in, file_in, url_site,
-                setting_state, history_state, model_state,
-                search_chk, language_dd, gr.State('auto')],
+        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]
     )
 
-    clr_btn.click(lambda: ([], [], "", []),
-                  outputs=[history_state, chat_out, preview_out, code_out])
+    clr_btn.click(
+        fn=lambda: ([], [], "", []),
+        outputs=[history_state, chat_out, preview_out, code_out]
+    )
 
 if __name__ == '__main__':
     demo.queue().launch()
-```