Upload 4 files

app.py

@@ -1,10 +1,11 @@
 import os
 from typing import Dict, List, Optional, Tuple
+import asyncio
 
 import gradio as gr
 
 from config import (
-    AVAILABLE_MODELS, DEMO_LIST, HTML_SYSTEM_PROMPT
+    AVAILABLE_MODELS, DEMO_LIST, HTML_SYSTEM_PROMPT,
 )
 from api_clients import generation_code, tavily_client
 from chat_processing import (
@@ -13,6 +14,10 @@ from chat_processing import (
 )
 from file_processing import create_multimodal_message
 from web_extraction import enhance_query_with_search
+from ux_components import create_demo_card, create_top_demo_cards
+
+# --- Gradio App UI ---
+
 
 
 def demo_card_click(e: gr.EventData):
@@ -41,7 +46,7 @@ def demo_card_click(e: gr.EventData):
         # Return the first demo description as fallback
         return DEMO_LIST[0]['description']
 
-# Main application
+
 with gr.Blocks(
     theme=gr.themes.Base(
         primary_hue="blue",
@@ -109,17 +114,8 @@ with gr.Blocks(
             visible=True  # Always visible
         )
         gr.Markdown("**Quick start**", visible=True)
-        with gr.Column(visible=True) as quick_examples_col:
-            for i, demo_item in enumerate(DEMO_LIST[:3]):
-                demo_card = gr.Button(
-                    value=demo_item['title'], 
-                    variant="secondary",
-                    size="sm"
-                )
-                demo_card.click(
-                    fn=lambda idx=i: gr.update(value=DEMO_LIST[idx]['description']),
-                    outputs=input
-                )
+        quick_examples_col = create_top_demo_cards(input)
+
         if not tavily_client:
             gr.Markdown("⚠️ Web search unavailable", visible=True)
         else:
@@ -130,6 +126,7 @@ with gr.Blocks(
                 if m['name'] == model_name:
                     return m, f"**Model:** {m['name']}", update_image_input_visibility(m)
             return AVAILABLE_MODELS[0], f"**Model:** {AVAILABLE_MODELS[0]['name']}", update_image_input_visibility(AVAILABLE_MODELS[0])  # Moonshot Kimi-K2 fallback
+
         def save_prompt(input):
             return {setting: {"system": input}}
         model_dropdown.change(
@@ -160,7 +157,7 @@ with gr.Blocks(
             with gr.Tab("History"):
                 history_output = gr.Chatbot(show_label=False, height=400, type="messages")
 
-    # Event handlers
+    # --- Event Handlers ---
     def update_code_language(language):
         return gr.update(language=get_gradio_language(language))
 
@@ -170,17 +167,30 @@ with gr.Blocks(
         if language == "html":
             return send_to_sandbox(code)
         else:
-            return "<div style='padding:1em;color:#888;text-align:center;'>Preview is only available for HTML. Please download your code using the download button above.</div>"
+            return "<div style='padding:1em;color:#888;text-align:center;'>Preview is only available for HTML.</div>"
+
+
+    # Use asyncio.create_task to ensure the generation_code coroutine is properly awaited
+    def submit_query(*args):
+        task = asyncio.create_task(generation_code(*args))  # Create a task
+        async def wrapper():
+            async for update in task:
+                yield update
+        return wrapper() # Return the wrapper function
+
+    btn.click(fn=submit_query,
+              inputs=[input, image_input, file_input, website_url_input, setting, history, current_model, search_toggle, language_dropdown],
+              outputs=[code_output, history, sandbox, history_output])
+
 
-    btn.click(
-        generation_code,
-        inputs=[input, image_input, file_input, website_url_input, setting, history, current_model, search_toggle, language_dropdown],
-        outputs=[code_output, history, sandbox, history_output]
-    )
     # Update preview when code or language changes
     code_output.change(preview_logic, inputs=[code_output, language_dropdown], outputs=sandbox)
     language_dropdown.change(preview_logic, inputs=[code_output, language_dropdown], outputs=sandbox)
     clear_btn.click(clear_history, outputs=[history, history_output, file_input, website_url_input])
 
+
+if __name__ == "__main__":
+    demo.queue(api_open=False, default_concurrency_limit=20).launch(ssr_mode=True, mcp_server=False, show_api=False)
+
 if __name__ == "__main__":
     demo.queue(api_open=False, default_concurrency_limit=20).launch(ssr_mode=True, mcp_server=False, show_api=False)

chat_processing.py

@@ -88,6 +88,10 @@ def update_image_input_visibility(model):
     is_glm_vl = model.get("id") == "THUDM/GLM-4.1V-9B-Thinking"
     return gr.update(visible=is_ernie_vl or is_glm_vl)
 
+def update_submit_button(query):
+    """Enable submit button if query is not empty"""
+    return gr.update(interactive=bool(query))
+
 def create_multimodal_message(text, image=None):
     """Create a multimodal message with text and optional image"""
     if image is None:

ux_components.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from config import DEMO_LIST
+
+def create_top_demo_cards(input_textbox):
+    """Creates a Gradio Column with buttons for the top 3 demo examples."""
+    with gr.Column(visible=True) as quick_examples_col:
+        for i, demo_item in enumerate(DEMO_LIST[:3]):
+            demo_card = gr.Button(
+                value=demo_item['title'],
+                variant="secondary",
+                size="sm",
+                elem_id=f"demo_card_{i}"  # Add an ID for potential styling
+            )
+            demo_card.click(
+                fn=lambda idx=i: gr.update(value=DEMO_LIST[idx]['description']),
+                outputs=input_textbox,
+            )
+    return quick_examples_col
+
+if __name__ == "__main__":
+    with gr.Blocks() as demo:
+        input_textbox = gr.Textbox(label="Input")
+        create_top_demo_cards(input_textbox)
+    demo.launch()

web_extraction.py

@@ -1,5 +1,5 @@
 import requests
-from urllib.parse import urlparse, urljoin
+from urllib.parse import urlparse, urljoin, ParseResult
 from bs4 import BeautifulSoup
 import re
 from tavily import TavilyClient
@@ -8,6 +8,7 @@ import os
 tavily_client = None
 TAVILY_API_KEY = os.getenv('TAVILY_API_KEY')
 if TAVILY_API_KEY:
+    import logging
     try:
         tavily_client = TavilyClient(api_key=TAVILY_API_KEY)
     except Exception as e:
@@ -74,6 +75,21 @@ def extract_website_content(url: str) -> str:
             return "Error: Invalid URL provided"
         
         # Set comprehensive headers to mimic a real browser request
+        scheme = parsed_url.scheme
+        netloc = parsed_url.netloc
+        path = parsed_url.path if parsed_url.path else "/"
+        params = parsed_url.params
+        query = parsed_url.query
+        fragment = parsed_url.fragment
+        reconstructed_url = ParseResult(scheme, netloc, path, params, query, fragment).geturl()
+
+        logging.info(f"Extracting content from: {reconstructed_url}")
+
+        if reconstructed_url != url:
+            logging.info(f"Original URL: {url}")
+            logging.info(f"Reconstructed URL: {reconstructed_url}")
+
+
         headers = {
             'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
             'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
@@ -99,7 +115,7 @@ def extract_website_content(url: str) -> str:
             try:
                 response = session.get(url, timeout=15, allow_redirects=True)
                 response.raise_for_status()
-                break
+                break # Exit the loop if successful
             except requests.exceptions.HTTPError as e:
                 if e.response.status_code == 403 and attempt < max_retries - 1:
                     # Try with different User-Agent on 403