Spaces:

natabrizy
/

testapp

Sleeping

App Files Files Community

natabrizy commited on Aug 20, 2025

Commit

46433dd

verified ·

1 Parent(s): 9aad0e4

Update app.py

Browse files

Files changed (1) hide show

app.py +72 -15

app.py CHANGED Viewed

@@ -16,9 +16,12 @@ from lzstring import LZString
 NEBIUS_BASE_URL = "https://api.studio.nebius.com/v1/"
 DEFAULT_VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
 DEFAULT_CODE_MODEL = "deepseek-ai/DeepSeek-V3-0324"
-HTTP_TIMEOUT = 60.0  # seconds
-# Use the same default key provided in your original code
 DEFAULT_NEBIUS_API_KEY = (
     "eyJhbGciOiJIUzI1NiIsImtpZCI6IlV6SXJWd1h0dnprLVRvdzlLZWstc0M1akptWXBvX1VaVkxUZlpnMDRlOFUiLCJ0eXAiOiJKV1QifQ.eyJzdWIiOiJnb29nbGUtb2F1dGgyfDEwNTA1MTQzMDg2MDMwMzIxNDEwMiIsInNjb3BlIjoib3BlbmlkIG9mZmxpbmVfYWNjZXNzIiwiaXNzIjoiYXBpX2tleV9pc3N1ZXIiLCJhdWQiOlsiaHR0cHM6Ly9uZWJpdXMtaW5mZXJlbmNlLmV1LmF1dGgwLmNvbS9hcGkvdjIvIl0sImV4cCI6MTkwNjU5ODA0NCwidXVpZCI6ImNkOGFiMWZlLTIxN2QtNDJlMy04OWUwLWM1YTg4MjcwMGVhNyIsIm5hbWUiOiJodW5nZ2luZyIsImV4cGlyZXNfYXQiOiIyMDMwLTA2LTAyVDAyOjM0OjA0KzAwMDAifQ.MA52QuIiNruK7_lX688RXAEI2TkcCOjcf_02XrpnhI8"
 )
@@ -32,7 +35,7 @@ def get_api_key(user_key: str = "") -> str:
     Resolve the Nebius API key from:
     1) The provided user_key field
     2) The NEBIUS_API_KEY environment variable
-    3) The built-in DEFAULT_NEBIUS_API_KEY (from your original code)
     """
     return (user_key or "").strip() or os.getenv("NEBIUS_API_KEY", "").strip() or DEFAULT_NEBIUS_API_KEY
@@ -46,7 +49,8 @@ def call_chat_completions(
 ) -> str:
     """
     Calls the Nebius OpenAI-compatible chat completions endpoint via HTTP.
-    Returns the message content string.
     """
     if not api_key:
         raise ValueError("Nebius API key is required.")
@@ -60,7 +64,10 @@ def call_chat_completions(
     }
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
-    with httpx.Client(timeout=HTTP_TIMEOUT) as client:
         resp = client.post(url, headers=headers, json=payload)
         resp.raise_for_status()
         data = resp.json()
@@ -143,9 +150,12 @@ def generate_html_code(
     description: str,
     nebius_api_key: str = "",
     code_model: str = DEFAULT_CODE_MODEL,
 ) -> str:
     """
     Generate HTML/CSS/JavaScript code based on a website description.
     """
     if not description or description.startswith("Error"):
         return "Error: Invalid or missing description."
@@ -177,17 +187,36 @@ Return only the complete HTML code starting with <!DOCTYPE html> and ending with
             model=code_model,
             messages=messages,
             api_key=api_key,
-            max_tokens=8000,
-            temperature=0.7,
         )
         html_code = _strip_fenced_code(content)
-        # Ensure HTML document bounds if present
         if "<!DOCTYPE html>" in html_code and "</html>" in html_code:
             start = html_code.find("<!DOCTYPE html>")
             end = html_code.rfind("</html>") + len("</html>")
             return html_code[start:end]
         return html_code
     except Exception as e:
         return f"Error generating HTML code: {str(e)}"
@@ -230,7 +259,8 @@ def create_codesandbox(html_code: str) -> str:
         # Try POST API to get a sandbox_id
         url = "https://codesandbox.io/api/v1/sandboxes/define"
-        with httpx.Client(timeout=HTTP_TIMEOUT) as client:
             resp = client.post(url, json=parameters)
             if resp.status_code == 200:
                 data = resp.json()
@@ -249,6 +279,8 @@ def screenshot_to_code(
     nebius_api_key: str = "",
     vision_model: str = DEFAULT_VISION_MODEL,
     code_model: str = DEFAULT_CODE_MODEL,
 ) -> Tuple[str, str]:
     """
     Complete pipeline: analyze image and generate corresponding HTML code.
@@ -257,7 +289,13 @@ def screenshot_to_code(
     description = analyze_image(image, nebius_api_key, vision_model)
     if description.startswith("Error"):
         return description, "Error: Cannot generate code due to image analysis failure."
-    html_code = generate_html_code(description, nebius_api_key, code_model)
     return description, html_code
@@ -277,7 +315,7 @@ def export_html_to_file(html_code: str) -> Optional[str]:
 # =========================
-# Gradio UI (clean, English-only)
 # =========================
 with gr.Blocks(
     theme=gr.themes.Soft(),
@@ -296,7 +334,7 @@ Turn website screenshots into functional HTML using Nebius models.
 - 📸 Image analysis (default: Qwen2.5-VL-72B-Instruct)
 - 💻 Code generation (default: DeepSeek-V3-0324)
 - 🌐 One-click CodeSandbox deployment
-- 🔒 Key stays on your machine; no storage
         """
     )
@@ -321,6 +359,23 @@ Turn website screenshots into functional HTML using Nebius models.
                 value=DEFAULT_CODE_MODEL,
                 allow_custom_value=True,
             )
     with gr.Tab("🎯 Quick Generate"):
         with gr.Row():
@@ -377,7 +432,7 @@ Turn website screenshots into functional HTML using Nebius models.
     # Event bindings
     generate_btn.click(
         fn=screenshot_to_code,
-        inputs=[image_input, nebius_key, vision_model_dd, code_model_dd],
         outputs=[description_output, html_output],
     )
@@ -410,8 +465,10 @@ Turn website screenshots into functional HTML using Nebius models.
     )
     code_btn.click(
-        fn=lambda desc, key, cmod: generate_html_code(desc, key, cmod),
-        inputs=[desc_input, nebius_key, code_model_dd],
         outputs=[code_result],
     )

 NEBIUS_BASE_URL = "https://api.studio.nebius.com/v1/"
 DEFAULT_VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
 DEFAULT_CODE_MODEL = "deepseek-ai/DeepSeek-V3-0324"
+# More forgiving timeouts and built-in retries to reduce "read operation timed out"
+HTTP_TIMEOUTS = httpx.Timeout(connect=10.0, read=120.0, write=30.0, pool=60.0)
+HTTP_RETRIES = 2  # extra attempts on transient failures
+# Use the same default key you provided
 DEFAULT_NEBIUS_API_KEY = (
     "eyJhbGciOiJIUzI1NiIsImtpZCI6IlV6SXJWd1h0dnprLVRvdzlLZWstc0M1akptWXBvX1VaVkxUZlpnMDRlOFUiLCJ0eXAiOiJKV1QifQ.eyJzdWIiOiJnb29nbGUtb2F1dGgyfDEwNTA1MTQzMDg2MDMwMzIxNDEwMiIsInNjb3BlIjoib3BlbmlkIG9mZmxpbmVfYWNjZXNzIiwiaXNzIjoiYXBpX2tleV9pc3N1ZXIiLCJhdWQiOlsiaHR0cHM6Ly9uZWJpdXMtaW5mZXJlbmNlLmV1LmF1dGgwLmNvbS9hcGkvdjIvIl0sImV4cCI6MTkwNjU5ODA0NCwidXVpZCI6ImNkOGFiMWZlLTIxN2QtNDJlMy04OWUwLWM1YTg4MjcwMGVhNyIsIm5hbWUiOiJodW5nZ2luZyIsImV4cGlyZXNfYXQiOiIyMDMwLTA2LTAyVDAyOjM0OjA0KzAwMDAifQ.MA52QuIiNruK7_lX688RXAEI2TkcCOjcf_02XrpnhI8"
 )
     Resolve the Nebius API key from:
     1) The provided user_key field
     2) The NEBIUS_API_KEY environment variable
+    3) The built-in DEFAULT_NEBIUS_API_KEY
     """
     return (user_key or "").strip() or os.getenv("NEBIUS_API_KEY", "").strip() or DEFAULT_NEBIUS_API_KEY
 ) -> str:
     """
     Calls the Nebius OpenAI-compatible chat completions endpoint via HTTP.
+    Returns the assistant message content string.
+    Includes retries and increased read timeout to mitigate timeouts.
     """
     if not api_key:
         raise ValueError("Nebius API key is required.")
     }
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+    # HTTPX transport with retries
+    transport = httpx.HTTPTransport(retries=HTTP_RETRIES)
+    with httpx.Client(timeout=HTTP_TIMEOUTS, transport=transport) as client:
         resp = client.post(url, headers=headers, json=payload)
         resp.raise_for_status()
         data = resp.json()
     description: str,
     nebius_api_key: str = "",
     code_model: str = DEFAULT_CODE_MODEL,
+    code_max_tokens: int = 4000,
+    code_temperature: float = 0.7,
 ) -> str:
     """
     Generate HTML/CSS/JavaScript code based on a website description.
+    Adds timeout-aware retry with a reduced token budget on second attempt.
     """
     if not description or description.startswith("Error"):
         return "Error: Invalid or missing description."
             model=code_model,
             messages=messages,
             api_key=api_key,
+            max_tokens=code_max_tokens,
+            temperature=code_temperature,
         )
         html_code = _strip_fenced_code(content)
         if "<!DOCTYPE html>" in html_code and "</html>" in html_code:
             start = html_code.find("<!DOCTYPE html>")
             end = html_code.rfind("</html>") + len("</html>")
             return html_code[start:end]
         return html_code
+    except (httpx.ReadTimeout, httpx.TimeoutException):
+        # Retry once with a reduced token budget to improve latency
+        try:
+            reduced_tokens = max(1000, int(code_max_tokens * 0.6))
+            messages = [{"role": "user", "content": prompt}]
+            content = call_chat_completions(
+                model=code_model,
+                messages=messages,
+                api_key=api_key,
+                max_tokens=reduced_tokens,
+                temperature=code_temperature,
+            )
+            html_code = _strip_fenced_code(content)
+            if "<!DOCTYPE html>" in html_code and "</html>" in html_code:
+                start = html_code.find("<!DOCTYPE html>")
+                end = html_code.rfind("</html>") + len("</html>")
+                return html_code[start:end]
+            return html_code
+        except Exception as e2:
+            return f"Error generating HTML code after retry: {str(e2)}. Tips: lower Max tokens, pick a smaller/faster model, or try again."
     except Exception as e:
         return f"Error generating HTML code: {str(e)}"
         # Try POST API to get a sandbox_id
         url = "https://codesandbox.io/api/v1/sandboxes/define"
+        transport = httpx.HTTPTransport(retries=HTTP_RETRIES)
+        with httpx.Client(timeout=HTTP_TIMEOUTS, transport=transport) as client:
             resp = client.post(url, json=parameters)
             if resp.status_code == 200:
                 data = resp.json()
     nebius_api_key: str = "",
     vision_model: str = DEFAULT_VISION_MODEL,
     code_model: str = DEFAULT_CODE_MODEL,
+    code_max_tokens: int = 4000,
+    code_temperature: float = 0.7,
 ) -> Tuple[str, str]:
     """
     Complete pipeline: analyze image and generate corresponding HTML code.
     description = analyze_image(image, nebius_api_key, vision_model)
     if description.startswith("Error"):
         return description, "Error: Cannot generate code due to image analysis failure."
+    html_code = generate_html_code(
+        description,
+        nebius_api_key,
+        code_model=code_model,
+        code_max_tokens=code_max_tokens,
+        code_temperature=code_temperature,
+    )
     return description, html_code
 # =========================
+# Gradio UI (English-only, user-friendly)
 # =========================
 with gr.Blocks(
     theme=gr.themes.Soft(),
 - 📸 Image analysis (default: Qwen2.5-VL-72B-Instruct)
 - 💻 Code generation (default: DeepSeek-V3-0324)
 - 🌐 One-click CodeSandbox deployment
+- 🔒 Your key is used at runtime only
         """
     )
                 value=DEFAULT_CODE_MODEL,
                 allow_custom_value=True,
             )
+        with gr.Row():
+            code_max_tokens = gr.Slider(
+                label="Max tokens (code generation)",
+                minimum=500,
+                maximum=8000,
+                step=100,
+                value=4000,
+                info="Lower this if you see timeouts; higher values may take longer.",
+            )
+            code_temperature = gr.Slider(
+                label="Temperature",
+                minimum=0.0,
+                maximum=1.5,
+                step=0.1,
+                value=0.7,
+                info="Higher is more creative; lower is more deterministic.",
+            )
     with gr.Tab("🎯 Quick Generate"):
         with gr.Row():
     # Event bindings
     generate_btn.click(
         fn=screenshot_to_code,
+        inputs=[image_input, nebius_key, vision_model_dd, code_model_dd, code_max_tokens, code_temperature],
         outputs=[description_output, html_output],
     )
     )
     code_btn.click(
+        fn=lambda desc, key, cmod, mtoks, temp: generate_html_code(
+            desc, key, code_model=cmod, code_max_tokens=mtoks, code_temperature=temp
+        ),
+        inputs=[desc_input, nebius_key, code_model_dd, code_max_tokens, code_temperature],
         outputs=[code_result],
     )