Upload server.py with huggingface_hub

server.py

@@ -42,7 +42,8 @@ def health():
 def generate():
     try:
         data = request.get_json(force=True)
-        prompt = data.get("inputs", "")
+        # Support both HF format (inputs) and backend format (prompt)
+        prompt = data.get("inputs", "") or data.get("prompt", "")
         params = data.get("parameters", {})
 
         width = int(params.get("width", 1024))
@@ -70,11 +71,45 @@ def generate():
         img_b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
         print(f"[image] Done, size: {len(buf.getvalue())} bytes", flush=True)
 
-        return jsonify({"image": img_b64, "format": "png"})
+        # Return in both formats - HF format and backend format
+        return jsonify({
+            "image": img_b64,
+            "format": "png",
+            "success": True,
+            "image_base64": img_b64,
+            "model": "glm-image",
+        })
     except Exception as e:
         tb = traceback.format_exc()
         print(f"[image] ERROR: {e}\n{tb}", flush=True)
-        return jsonify({"error": str(e), "traceback": tb}), 500
+        return jsonify({"error": str(e), "traceback": tb, "success": False}), 500
+
+# Also add /api/generate route for the backend's expected format
+@app.route("/api/generate", methods=["POST"])
+def api_generate():
+    """Handle the backend's /api/generate format by forwarding to generate()"""
+    return generate()
+
+# Also add /api/analyze and /api/vision-chat for the backend
+@app.route("/api/analyze", methods=["POST"])
+def api_analyze():
+    """Image analysis - return a placeholder since we don't have a vision model here"""
+    data = request.get_json(force=True)
+    return jsonify({
+        "success": True,
+        "description": "Image analysis is not available on this endpoint.",
+        "model": "none",
+    })
+
+@app.route("/api/vision-chat", methods=["POST"])
+def api_vision_chat():
+    """Vision chat - return a placeholder"""
+    data = request.get_json(force=True)
+    return jsonify({
+        "success": True,
+        "answer": "Vision chat is not available on this endpoint.",
+        "model": "none",
+    })
 
 if __name__ == "__main__":
     load_model()