change to Nebius API

app.py

@@ -6,21 +6,21 @@ from PIL import Image
 import httpx
 import html
 from lzstring import LZString
-import google.generativeai as genai
+import os
+from openai import OpenAI
 
 # --- 默认API配置 ---
-DEFAULT_GEMINI_API_KEY = ""
-DEFAULT_DEEPSEEK_API_KEY = ""
-DEEPSEEK_BASE_URL = "https://api.deepseek.com"
+DEFAULT_NEBIUS_API_KEY = "eyJhbGciOiJIUzI1NiIsImtpZCI6IlV6SXJWd1h0dnprLVRvdzlLZWstc0M1akptWXBvX1VaVkxUZlpnMDRlOFUiLCJ0eXAiOiJKV1QifQ.eyJzdWIiOiJnb29nbGUtb2F1dGgyfDEwNTA1MTQzMDg2MDMwMzIxNDEwMiIsInNjb3BlIjoib3BlbmlkIG9mZmxpbmVfYWNjZXNzIiwiaXNzIjoiYXBpX2tleV9pc3N1ZXIiLCJhdWQiOlsiaHR0cHM6Ly9uZWJpdXMtaW5mZXJlbmNlLmV1LmF1dGgwLmNvbS9hcGkvdjIvIl0sImV4cCI6MTkwNjU5ODA0NCwidXVpZCI6ImNkOGFiMWZlLTIxN2QtNDJlMy04OWUwLWM1YTg4MjcwMGVhNyIsIm5hbWUiOiJodW5nZ2luZyIsImV4cGlyZXNfYXQiOiIyMDMwLTA2LTAyVDAyOjM0OjA0KzAwMDAifQ.MA52QuIiNruK7_lX688RXAEI2TkcCOjcf_02XrpnhI8"
+NEBIUS_BASE_URL = "https://api.studio.nebius.com/v1/"
 
 # --- 核心工具函数 ---
-def analyze_image(image: Image.Image, gemini_api_key: str = "") -> str:
+def analyze_image(image: Image.Image, nebius_api_key: str = "") -> str:
     """
     Analyze an uploaded image and provide a detailed description of its content and layout.
     
     Args:
         image: The PIL Image object to analyze
-        gemini_api_key: Gemini API key for image analysis
+        nebius_api_key: Nebius API key for image analysis
         
     Returns:
         A detailed description of the image content, layout, and website type
@@ -29,22 +29,24 @@ def analyze_image(image: Image.Image, gemini_api_key: str = "") -> str:
         return "Error: No image provided"
     
     # 使用提供的API密钥或默认密钥
-    api_key = gemini_api_key.strip() if gemini_api_key.strip() else DEFAULT_GEMINI_API_KEY
+    api_key = nebius_api_key.strip() if nebius_api_key.strip() else DEFAULT_NEBIUS_API_KEY
     
     if not api_key:
-        return "Error: Gemini API key not provided"
+        return "Error: Nebius API key not provided"
     
     try:
-        # 配置Gemini API
-        genai.configure(api_key=api_key)
+        # 配置Nebius OpenAI客户端
+        client = OpenAI(
+            base_url=NEBIUS_BASE_URL,
+            api_key=api_key,
+        )
         
         # 转换图片为base64
         buffered = io.BytesIO()
         image.save(buffered, format="PNG")
         img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
         
-        # 创建模型和提示
-        model = genai.GenerativeModel(model_name="gemini-2.5-flash-preview-05-20")
+        # 创建提示
         prompt = """
         Analyze this image and provide a concise description.
         Describe the main elements, colors, layout, and UI components.
@@ -52,23 +54,39 @@ def analyze_image(image: Image.Image, gemini_api_key: str = "") -> str:
         Focus on structural and visual elements that would be important for recreating the design.
         """
         
-        image_part = {"mime_type": "image/png", "data": img_str}
-        contents = [prompt, image_part]
+        # 使用Qwen2.5-VL-72B-Instruct模型进行图像分析
+        response = client.chat.completions.create(
+            model="Qwen/Qwen2.5-VL-72B-Instruct",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": prompt},
+                        {
+                            "type": "image_url",
+                            "image_url": {
+                                "url": f"data:image/png;base64,{img_str}"
+                            }
+                        }
+                    ]
+                }
+            ],
+            max_tokens=1000,
+            temperature=0.7
+        )
         
-        # 生成描述
-        response = model.generate_content(contents)
-        return response.text
+        return response.choices[0].message.content
         
     except Exception as e:
         return f"Error analyzing image: {str(e)}"
 
-def generate_html_code(description: str, deepseek_api_key: str = "") -> str:
+def generate_html_code(description: str, nebius_api_key: str = "") -> str:
     """
     Generate HTML/CSS/JavaScript code based on a website description.
     
     Args:
         description: Detailed description of the website to generate
-        deepseek_api_key: DeepSeek API key for code generation
+        nebius_api_key: Nebius API key for code generation
         
     Returns:
         Complete HTML code with embedded CSS and JavaScript
@@ -77,10 +95,10 @@ def generate_html_code(description: str, deepseek_api_key: str = "") -> str:
         return "Error: Invalid or missing description"
     
     # 使用提供的API密钥或默认密钥
-    api_key = deepseek_api_key.strip() if deepseek_api_key.strip() else DEFAULT_DEEPSEEK_API_KEY
+    api_key = nebius_api_key.strip() if nebius_api_key.strip() else DEFAULT_NEBIUS_API_KEY
     
     if not api_key:
-        return "Error: DeepSeek API key not provided"
+        return "Error: Nebius API key not provided"
     
     prompt = f"""
     Generate a complete, responsive webpage based on this description:
@@ -99,46 +117,38 @@ def generate_html_code(description: str, deepseek_api_key: str = "") -> str:
     Return only the complete HTML code starting with <!DOCTYPE html> and ending with </html>.
     """
     
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    
-    data = {
-        "model": "deepseek-chat",
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": 0.7,
-        "max_tokens": 8000
-    }
-    
     try:
-        import requests
-        response = requests.post(
-            f"{DEEPSEEK_BASE_URL}/chat/completions",
-            headers=headers,
-            json=data,
-            timeout=60
+        # 配置Nebius OpenAI客户端
+        client = OpenAI(
+            base_url=NEBIUS_BASE_URL,
+            api_key=api_key,
         )
         
-        if response.status_code == 200:
-            result = response.json()
-            html_code = result['choices'][0]['message']['content']
-            
-            # 清理代码格式
-            if html_code.strip().startswith("```html"):
-                html_code = html_code.split("```html", 1)[1].strip()
-            if html_code.strip().endswith("```"):
-                html_code = html_code.rsplit("```", 1)[0].strip()
-            
-            # 确保代码完整性
-            if "<!DOCTYPE html>" in html_code and "</html>" in html_code:
-                start = html_code.find("<!DOCTYPE html>")
-                end = html_code.rfind("</html>") + 7
-                return html_code[start:end]
-            else:
-                return html_code
+        # 使用DeepSeek-V3-0324模型进行代码生成
+        response = client.chat.completions.create(
+            model="deepseek-ai/DeepSeek-V3-0324",
+            messages=[
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=8000,
+            temperature=0.7
+        )
+        
+        html_code = response.choices[0].message.content
+        
+        # 清理代码格式
+        if html_code.strip().startswith("```html"):
+            html_code = html_code.split("```html", 1)[1].strip()
+        if html_code.strip().endswith("```"):
+            html_code = html_code.rsplit("```", 1)[0].strip()
+        
+        # 确保代码完整性
+        if "<!DOCTYPE html>" in html_code and "</html>" in html_code:
+            start = html_code.find("<!DOCTYPE html>")
+            end = html_code.rfind("</html>") + 7
+            return html_code[start:end]
         else:
-            return f"Error: API request failed with status {response.status_code}"
+            return html_code
             
     except Exception as e:
         return f"Error generating HTML code: {str(e)}"
@@ -198,7 +208,17 @@ def create_codesandbox(html_code: str) -> str:
         
         # 尝试创建sandbox
         import requests
-        response = requests.post(codesandbox_url, timeout=30)
+        
+        # 尝试POST请求创建sandbox
+        try:
+            response = requests.post(
+                "https://codesandbox.io/api/v1/sandboxes/define",
+                json=parameters,
+                timeout=10
+            )
+        except:
+            # 如果POST失败，返回GET URL
+            return codesandbox_url
         
         if response.status_code == 200:
             result = response.json()
@@ -212,33 +232,32 @@ def create_codesandbox(html_code: str) -> str:
     except Exception as e:
         return f"Error creating CodeSandbox: {str(e)}"
 
-def screenshot_to_code(image: Image.Image, gemini_api_key: str = "", deepseek_api_key: str = "") -> tuple:
+def screenshot_to_code(image: Image.Image, nebius_api_key: str = "") -> tuple:
     """
     Complete pipeline: analyze image and generate corresponding HTML code.
     
     Args:
         image: Screenshot image to analyze
-        gemini_api_key: Gemini API key for image analysis
-        deepseek_api_key: DeepSeek API key for code generation
+        nebius_api_key: Nebius API key for both image analysis and code generation
         
     Returns:
         Tuple of (description, html_code)
     """
     # 分析图片
-    description = analyze_image(image, gemini_api_key)
+    description = analyze_image(image, nebius_api_key)
     
     if description.startswith("Error"):
         return description, "Error: Cannot generate code due to image analysis failure"
     
     # 生成代码
-    html_code = generate_html_code(description, deepseek_api_key)
+    html_code = generate_html_code(description, nebius_api_key)
     
     return description, html_code
 
 # --- Gradio界面 ---
 with gr.Blocks(
     theme=gr.themes.Soft(),
-    title="AI Website Generator - MCP Compatible",
+    title="AI Website Generator - MCP Compatible (Nebius)",
     css="""
     .api-section { background: #f8f9fa; padding: 15px; border-radius: 8px; margin: 10px 0; }
     .tool-section { border: 1px solid #e0e0e0; padding: 15px; border-radius: 8px; margin: 10px 0; }
@@ -246,13 +265,13 @@ with gr.Blocks(
 ) as app:
     
     gr.Markdown("""
-    # 🚀 AI Website Generator (MCP Compatible)
+    # 🚀 AI Website Generator (MCP Compatible - Nebius)
     
-    Transform website screenshots into functional HTML code using AI.
+    Transform website screenshots into functional HTML code using Nebius AI.
     
     **Features:**
-    - 📸 Image analysis with Gemini AI
-    - 💻 HTML/CSS/JS code generation with DeepSeek
+    - 📸 Image analysis with Qwen2.5-VL-72B-Instruct
+    - 💻 HTML/CSS/JS code generation with DeepSeek-V3-0324
     - 🌐 Direct CodeSandbox deployment
     - 🔧 MCP (Model Context Protocol) compatible
     
@@ -270,17 +289,11 @@ with gr.Blocks(
                 
                 # API配置
                 with gr.Group():
-                    gr.Markdown("**API Keys (Optional)**")
-                    gemini_key = gr.Textbox(
-                        label="Gemini API Key",
-                        type="password",
-                        placeholder="Leave empty to use default",
-                        value=""
-                    )
-                    deepseek_key = gr.Textbox(
-                        label="DeepSeek API Key",
+                    gr.Markdown("**Nebius API Key (Required)**")
+                    nebius_key = gr.Textbox(
+                        label="Nebius API Key",
                         type="password",
-                        placeholder="Leave empty to use default",
+                        placeholder="Enter your Nebius API key",
                         value=""
                     )
                 
@@ -326,21 +339,21 @@ with gr.Blocks(
             with gr.Column():
                 gr.Markdown("#### 📸 Image Analysis Tool")
                 img_tool = gr.Image(type="pil", label="Image")
-                gemini_key_tool = gr.Textbox(label="Gemini API Key", type="password")
+                nebius_key_tool = gr.Textbox(label="Nebius API Key", type="password")
                 analyze_btn = gr.Button("Analyze Image")
                 analysis_result = gr.Textbox(label="Analysis Result", lines=5)
             
             with gr.Column():
                 gr.Markdown("#### 💻 Code Generation Tool")
                 desc_input = gr.Textbox(label="Description", lines=3)
-                deepseek_key_tool = gr.Textbox(label="DeepSeek API Key", type="password")
+                nebius_key_tool2 = gr.Textbox(label="Nebius API Key", type="password")
                 code_btn = gr.Button("Generate Code")
                 code_result = gr.Code(label="Generated Code", language="html")
     
     # 事件绑定
     generate_btn.click(
         fn=screenshot_to_code,
-        inputs=[image_input, gemini_key, deepseek_key],
+        inputs=[image_input, nebius_key],
         outputs=[description_output, html_output]
     )
     
@@ -352,13 +365,13 @@ with gr.Blocks(
     
     analyze_btn.click(
         fn=analyze_image,
-        inputs=[img_tool, gemini_key_tool],
+        inputs=[img_tool, nebius_key_tool],
         outputs=[analysis_result]
     )
     
     code_btn.click(
         fn=generate_html_code,
-        inputs=[desc_input, deepseek_key_tool],
+        inputs=[desc_input, nebius_key_tool2],
         outputs=[code_result]
     )
     

requirements.txt

@@ -1,6 +1,6 @@
 gradio>=4.19.0
 pillow>=9.0.0
 httpx>=0.24.0
-google-generativeai>=0.3.0
 lzstring>=1.0.4
-requests>=2.28.0
+requests>=2.28.0
+openai>=1.0.0